├── .gitignore ├── LICENSE ├── README.md ├── assets └── teaser.png ├── contrib └── conda_env.yaml ├── data ├── __init__.py ├── aligned_dataset.py ├── base_dataset.py ├── colorization_dataset.py ├── copy_dataset.py ├── filter_blurimg.py ├── image_folder.py ├── single_dataset.py ├── template_dataset.py └── unaligned_dataset.py ├── dataset └── download_RB2V.sh ├── models ├── __init__.py ├── backbone │ ├── arch_util.py │ ├── local_arch.py │ ├── netD.py │ ├── netE.py │ └── netG.py ├── base_model.py ├── blur2blur_model.py ├── explore │ ├── __init__.py │ ├── arch_util.py │ ├── backbones │ │ ├── __pycache__ │ │ │ ├── resnet.cpython-38.pyc │ │ │ ├── resnet.cpython-39.pyc │ │ │ ├── unet_parts.cpython-38.pyc │ │ │ └── unet_parts.cpython-39.pyc │ │ ├── resnet.py │ │ ├── skip │ │ │ ├── concat.py │ │ │ ├── downsampler.py │ │ │ ├── non_local_dot_product.py │ │ │ ├── skip.py │ │ │ └── util.py │ │ └── unet_parts.py │ ├── deblurring │ │ ├── image_deblur.py │ │ └── joint_deblur.py │ ├── dips.py │ ├── kernel_encoding │ │ ├── base_model.py │ │ ├── image_base_model.py │ │ └── kernel_wizard.py │ ├── losses │ │ ├── charbonnier_loss.py │ │ ├── dsd_loss.py │ │ ├── gan_loss.py │ │ ├── hyper_laplacian_penalty.py │ │ ├── perceptual_loss.py │ │ └── ssim_loss.py │ └── lr_scheduler.py ├── losses │ ├── __init__.py │ ├── gan_loss.py │ ├── lpips_loss.py │ └── vgg_loss.py ├── mimo │ ├── MIMOUNet.py │ └── layers.py ├── networks.py ├── pg_modules │ ├── __pycache__ │ │ ├── blocks.cpython-39.pyc │ │ ├── diffaug.cpython-39.pyc │ │ ├── discriminator.cpython-39.pyc │ │ └── projector.cpython-39.pyc │ ├── blocks.py │ ├── diffaug.py │ ├── discriminator.py │ ├── networks_fastgan.py │ ├── networks_stylegan2.py │ └── projector.py ├── pretrained_networks.py ├── template_model.py ├── test_model.py ├── unet_parts.py └── weights │ └── v0.1 │ └── vgg.pth ├── options ├── __init__.py ├── base_options.py ├── generate_blur │ ├── augmentation.yml │ └── default.yml ├── test_options.py └── train_options.py ├── psnr_ssim.py ├── requirements.txt ├── scripts ├── test.sh └── train.sh ├── setup.py ├── test.py ├── tests ├── __init__.py └── test_queues.py ├── train.py └── util ├── __init__.py ├── get_data.py ├── html.py ├── image_pool.py ├── psnr_ssim.py ├── util.py └── visualizer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/README.md -------------------------------------------------------------------------------- /assets/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/assets/teaser.png -------------------------------------------------------------------------------- /contrib/conda_env.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/contrib/conda_env.yaml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/aligned_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/data/aligned_dataset.py -------------------------------------------------------------------------------- /data/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/data/base_dataset.py -------------------------------------------------------------------------------- /data/colorization_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/data/colorization_dataset.py -------------------------------------------------------------------------------- /data/copy_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/data/copy_dataset.py -------------------------------------------------------------------------------- /data/filter_blurimg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/data/filter_blurimg.py -------------------------------------------------------------------------------- /data/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/data/image_folder.py -------------------------------------------------------------------------------- /data/single_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/data/single_dataset.py -------------------------------------------------------------------------------- /data/template_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/data/template_dataset.py -------------------------------------------------------------------------------- /data/unaligned_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/data/unaligned_dataset.py -------------------------------------------------------------------------------- /dataset/download_RB2V.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/dataset/download_RB2V.sh -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/backbone/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/backbone/arch_util.py -------------------------------------------------------------------------------- /models/backbone/local_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/backbone/local_arch.py -------------------------------------------------------------------------------- /models/backbone/netD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/backbone/netD.py -------------------------------------------------------------------------------- /models/backbone/netE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/backbone/netE.py -------------------------------------------------------------------------------- /models/backbone/netG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/backbone/netG.py -------------------------------------------------------------------------------- /models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/base_model.py -------------------------------------------------------------------------------- /models/blur2blur_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/blur2blur_model.py -------------------------------------------------------------------------------- /models/explore/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/__init__.py -------------------------------------------------------------------------------- /models/explore/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/arch_util.py -------------------------------------------------------------------------------- /models/explore/backbones/__pycache__/resnet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/backbones/__pycache__/resnet.cpython-38.pyc -------------------------------------------------------------------------------- /models/explore/backbones/__pycache__/resnet.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/backbones/__pycache__/resnet.cpython-39.pyc -------------------------------------------------------------------------------- /models/explore/backbones/__pycache__/unet_parts.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/backbones/__pycache__/unet_parts.cpython-38.pyc -------------------------------------------------------------------------------- /models/explore/backbones/__pycache__/unet_parts.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/backbones/__pycache__/unet_parts.cpython-39.pyc -------------------------------------------------------------------------------- /models/explore/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/backbones/resnet.py -------------------------------------------------------------------------------- /models/explore/backbones/skip/concat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/backbones/skip/concat.py -------------------------------------------------------------------------------- /models/explore/backbones/skip/downsampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/backbones/skip/downsampler.py -------------------------------------------------------------------------------- /models/explore/backbones/skip/non_local_dot_product.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/backbones/skip/non_local_dot_product.py -------------------------------------------------------------------------------- /models/explore/backbones/skip/skip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/backbones/skip/skip.py -------------------------------------------------------------------------------- /models/explore/backbones/skip/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/backbones/skip/util.py -------------------------------------------------------------------------------- /models/explore/backbones/unet_parts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/backbones/unet_parts.py -------------------------------------------------------------------------------- /models/explore/deblurring/image_deblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/deblurring/image_deblur.py -------------------------------------------------------------------------------- /models/explore/deblurring/joint_deblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/deblurring/joint_deblur.py -------------------------------------------------------------------------------- /models/explore/dips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/dips.py -------------------------------------------------------------------------------- /models/explore/kernel_encoding/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/kernel_encoding/base_model.py -------------------------------------------------------------------------------- /models/explore/kernel_encoding/image_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/kernel_encoding/image_base_model.py -------------------------------------------------------------------------------- /models/explore/kernel_encoding/kernel_wizard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/kernel_encoding/kernel_wizard.py -------------------------------------------------------------------------------- /models/explore/losses/charbonnier_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/losses/charbonnier_loss.py -------------------------------------------------------------------------------- /models/explore/losses/dsd_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/losses/dsd_loss.py -------------------------------------------------------------------------------- /models/explore/losses/gan_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/losses/gan_loss.py -------------------------------------------------------------------------------- /models/explore/losses/hyper_laplacian_penalty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/losses/hyper_laplacian_penalty.py -------------------------------------------------------------------------------- /models/explore/losses/perceptual_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/losses/perceptual_loss.py -------------------------------------------------------------------------------- /models/explore/losses/ssim_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/losses/ssim_loss.py -------------------------------------------------------------------------------- /models/explore/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/explore/lr_scheduler.py -------------------------------------------------------------------------------- /models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/losses/__init__.py -------------------------------------------------------------------------------- /models/losses/gan_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/losses/gan_loss.py -------------------------------------------------------------------------------- /models/losses/lpips_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/losses/lpips_loss.py -------------------------------------------------------------------------------- /models/losses/vgg_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/losses/vgg_loss.py -------------------------------------------------------------------------------- /models/mimo/MIMOUNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/mimo/MIMOUNet.py -------------------------------------------------------------------------------- /models/mimo/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/mimo/layers.py -------------------------------------------------------------------------------- /models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/networks.py -------------------------------------------------------------------------------- /models/pg_modules/__pycache__/blocks.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/pg_modules/__pycache__/blocks.cpython-39.pyc -------------------------------------------------------------------------------- /models/pg_modules/__pycache__/diffaug.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/pg_modules/__pycache__/diffaug.cpython-39.pyc -------------------------------------------------------------------------------- /models/pg_modules/__pycache__/discriminator.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/pg_modules/__pycache__/discriminator.cpython-39.pyc -------------------------------------------------------------------------------- /models/pg_modules/__pycache__/projector.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/pg_modules/__pycache__/projector.cpython-39.pyc -------------------------------------------------------------------------------- /models/pg_modules/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/pg_modules/blocks.py -------------------------------------------------------------------------------- /models/pg_modules/diffaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/pg_modules/diffaug.py -------------------------------------------------------------------------------- /models/pg_modules/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/pg_modules/discriminator.py -------------------------------------------------------------------------------- /models/pg_modules/networks_fastgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/pg_modules/networks_fastgan.py -------------------------------------------------------------------------------- /models/pg_modules/networks_stylegan2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/pg_modules/networks_stylegan2.py -------------------------------------------------------------------------------- /models/pg_modules/projector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/pg_modules/projector.py -------------------------------------------------------------------------------- /models/pretrained_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/pretrained_networks.py -------------------------------------------------------------------------------- /models/template_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/template_model.py -------------------------------------------------------------------------------- /models/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/test_model.py -------------------------------------------------------------------------------- /models/unet_parts.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/weights/v0.1/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/models/weights/v0.1/vgg.pth -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/options/__init__.py -------------------------------------------------------------------------------- /options/base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/options/base_options.py -------------------------------------------------------------------------------- /options/generate_blur/augmentation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/options/generate_blur/augmentation.yml -------------------------------------------------------------------------------- /options/generate_blur/default.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/options/generate_blur/default.yml -------------------------------------------------------------------------------- /options/test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/options/test_options.py -------------------------------------------------------------------------------- /options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/options/train_options.py -------------------------------------------------------------------------------- /psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/psnr_ssim.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/scripts/test.sh -------------------------------------------------------------------------------- /scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/scripts/train.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/setup.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/test.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_queues.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/tests/test_queues.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/train.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/util/__init__.py -------------------------------------------------------------------------------- /util/get_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/util/get_data.py -------------------------------------------------------------------------------- /util/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/util/html.py -------------------------------------------------------------------------------- /util/image_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/util/image_pool.py -------------------------------------------------------------------------------- /util/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/util/psnr_ssim.py -------------------------------------------------------------------------------- /util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/util/util.py -------------------------------------------------------------------------------- /util/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/Blur2Blur/HEAD/util/visualizer.py --------------------------------------------------------------------------------