├── LICENSE ├── README.md ├── codes ├── auto_test.py ├── data │ ├── LRHR_dataset.py │ ├── LRHR_seg_bg_dataset.py │ ├── LR_dataset.py │ ├── __init__.py │ └── util.py ├── models │ ├── SFTGAN_ACD_model.py │ ├── SRGAN_model.py │ ├── SRRaGAN_model.py │ ├── SR_model.py │ ├── __init__.py │ ├── base_model.py │ ├── modules │ │ ├── architecture.py │ │ ├── block.py │ │ ├── loss.py │ │ ├── seg_arch.py │ │ ├── sft_arch.py │ │ └── spectral_norm.py │ └── networks.py ├── options │ ├── options.py │ ├── test │ │ ├── test_ESRGANplus.json │ │ ├── test_SRGAN.json │ │ ├── test_SRResNet.json │ │ └── test_sr.json │ └── train │ │ ├── train_ESRGANplus.json │ │ ├── train_SRGAN.json │ │ ├── train_SRResNet.json │ │ ├── train_sftgan.json │ │ └── train_sr.json ├── scripts │ ├── README.md │ ├── back_projection │ │ ├── backprojection.m │ │ ├── main_bp.m │ │ └── main_reverse_filter.m │ ├── color2gray.py │ ├── create_lmdb.py │ ├── extract_enlarge_patches.py │ ├── extract_subimgs_single.py │ ├── generate_mod_LR_bic.m │ ├── generate_mod_LR_bic.py │ ├── make_gif_video.py │ ├── net_interp.py │ ├── rename.py │ ├── transfer_params.py │ └── transfer_params_sft.py ├── test.py ├── test_dataloader.py ├── test_seg.py ├── test_sftgan.py ├── train.py └── utils │ ├── progress_bar.py │ └── util.py ├── experiments └── pretrained_models │ └── README.md ├── figures ├── noise_per_residual_dense_block.PNG └── qualitative_result.PNG ├── metrics ├── calculate_PSNR_SSIM.m └── calculate_PSNR_SSIM.py └── test_image ├── LR ├── baby.png ├── bird.png ├── butterfly.png ├── head.png └── woman.png ├── architecture.py ├── block.py ├── pretrained_models └── README.md ├── results ├── baby_nESRGANplus.png ├── bird_nESRGANplus.png ├── butterfly_nESRGANplus.png ├── head_nESRGANplus.png └── woman_nESRGANplus.png └── test.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/README.md -------------------------------------------------------------------------------- /codes/auto_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/auto_test.py -------------------------------------------------------------------------------- /codes/data/LRHR_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/data/LRHR_dataset.py -------------------------------------------------------------------------------- /codes/data/LRHR_seg_bg_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/data/LRHR_seg_bg_dataset.py -------------------------------------------------------------------------------- /codes/data/LR_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/data/LR_dataset.py -------------------------------------------------------------------------------- /codes/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/data/__init__.py -------------------------------------------------------------------------------- /codes/data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/data/util.py -------------------------------------------------------------------------------- /codes/models/SFTGAN_ACD_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/models/SFTGAN_ACD_model.py -------------------------------------------------------------------------------- /codes/models/SRGAN_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/models/SRGAN_model.py -------------------------------------------------------------------------------- /codes/models/SRRaGAN_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/models/SRRaGAN_model.py -------------------------------------------------------------------------------- /codes/models/SR_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/models/SR_model.py -------------------------------------------------------------------------------- /codes/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/models/__init__.py -------------------------------------------------------------------------------- /codes/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/models/base_model.py -------------------------------------------------------------------------------- /codes/models/modules/architecture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/models/modules/architecture.py -------------------------------------------------------------------------------- /codes/models/modules/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/models/modules/block.py -------------------------------------------------------------------------------- /codes/models/modules/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/models/modules/loss.py -------------------------------------------------------------------------------- /codes/models/modules/seg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/models/modules/seg_arch.py -------------------------------------------------------------------------------- /codes/models/modules/sft_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/models/modules/sft_arch.py -------------------------------------------------------------------------------- /codes/models/modules/spectral_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/models/modules/spectral_norm.py -------------------------------------------------------------------------------- /codes/models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/models/networks.py -------------------------------------------------------------------------------- /codes/options/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/options/options.py -------------------------------------------------------------------------------- /codes/options/test/test_ESRGANplus.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/options/test/test_ESRGANplus.json -------------------------------------------------------------------------------- /codes/options/test/test_SRGAN.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/options/test/test_SRGAN.json -------------------------------------------------------------------------------- /codes/options/test/test_SRResNet.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/options/test/test_SRResNet.json -------------------------------------------------------------------------------- /codes/options/test/test_sr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/options/test/test_sr.json -------------------------------------------------------------------------------- /codes/options/train/train_ESRGANplus.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/options/train/train_ESRGANplus.json -------------------------------------------------------------------------------- /codes/options/train/train_SRGAN.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/options/train/train_SRGAN.json -------------------------------------------------------------------------------- /codes/options/train/train_SRResNet.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/options/train/train_SRResNet.json -------------------------------------------------------------------------------- /codes/options/train/train_sftgan.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/options/train/train_sftgan.json -------------------------------------------------------------------------------- /codes/options/train/train_sr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/options/train/train_sr.json -------------------------------------------------------------------------------- /codes/scripts/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/README.md -------------------------------------------------------------------------------- /codes/scripts/back_projection/backprojection.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/back_projection/backprojection.m -------------------------------------------------------------------------------- /codes/scripts/back_projection/main_bp.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/back_projection/main_bp.m -------------------------------------------------------------------------------- /codes/scripts/back_projection/main_reverse_filter.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/back_projection/main_reverse_filter.m -------------------------------------------------------------------------------- /codes/scripts/color2gray.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/color2gray.py -------------------------------------------------------------------------------- /codes/scripts/create_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/create_lmdb.py -------------------------------------------------------------------------------- /codes/scripts/extract_enlarge_patches.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/extract_enlarge_patches.py -------------------------------------------------------------------------------- /codes/scripts/extract_subimgs_single.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/extract_subimgs_single.py -------------------------------------------------------------------------------- /codes/scripts/generate_mod_LR_bic.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/generate_mod_LR_bic.m -------------------------------------------------------------------------------- /codes/scripts/generate_mod_LR_bic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/generate_mod_LR_bic.py -------------------------------------------------------------------------------- /codes/scripts/make_gif_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/make_gif_video.py -------------------------------------------------------------------------------- /codes/scripts/net_interp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/net_interp.py -------------------------------------------------------------------------------- /codes/scripts/rename.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/rename.py -------------------------------------------------------------------------------- /codes/scripts/transfer_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/transfer_params.py -------------------------------------------------------------------------------- /codes/scripts/transfer_params_sft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/scripts/transfer_params_sft.py -------------------------------------------------------------------------------- /codes/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/test.py -------------------------------------------------------------------------------- /codes/test_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/test_dataloader.py -------------------------------------------------------------------------------- /codes/test_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/test_seg.py -------------------------------------------------------------------------------- /codes/test_sftgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/test_sftgan.py -------------------------------------------------------------------------------- /codes/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/train.py -------------------------------------------------------------------------------- /codes/utils/progress_bar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/utils/progress_bar.py -------------------------------------------------------------------------------- /codes/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/codes/utils/util.py -------------------------------------------------------------------------------- /experiments/pretrained_models/README.md: -------------------------------------------------------------------------------- 1 | # Pretrained models -------------------------------------------------------------------------------- /figures/noise_per_residual_dense_block.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/figures/noise_per_residual_dense_block.PNG -------------------------------------------------------------------------------- /figures/qualitative_result.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/figures/qualitative_result.PNG -------------------------------------------------------------------------------- /metrics/calculate_PSNR_SSIM.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/metrics/calculate_PSNR_SSIM.m -------------------------------------------------------------------------------- /metrics/calculate_PSNR_SSIM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/metrics/calculate_PSNR_SSIM.py -------------------------------------------------------------------------------- /test_image/LR/baby.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/test_image/LR/baby.png -------------------------------------------------------------------------------- /test_image/LR/bird.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/test_image/LR/bird.png -------------------------------------------------------------------------------- /test_image/LR/butterfly.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/test_image/LR/butterfly.png -------------------------------------------------------------------------------- /test_image/LR/head.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/test_image/LR/head.png -------------------------------------------------------------------------------- /test_image/LR/woman.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/test_image/LR/woman.png -------------------------------------------------------------------------------- /test_image/architecture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/test_image/architecture.py -------------------------------------------------------------------------------- /test_image/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/test_image/block.py -------------------------------------------------------------------------------- /test_image/pretrained_models/README.md: -------------------------------------------------------------------------------- 1 | # Place pretrained models here -------------------------------------------------------------------------------- /test_image/results/baby_nESRGANplus.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/test_image/results/baby_nESRGANplus.png -------------------------------------------------------------------------------- /test_image/results/bird_nESRGANplus.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/test_image/results/bird_nESRGANplus.png -------------------------------------------------------------------------------- /test_image/results/butterfly_nESRGANplus.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/test_image/results/butterfly_nESRGANplus.png -------------------------------------------------------------------------------- /test_image/results/head_nESRGANplus.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/test_image/results/head_nESRGANplus.png -------------------------------------------------------------------------------- /test_image/results/woman_nESRGANplus.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/test_image/results/woman_nESRGANplus.png -------------------------------------------------------------------------------- /test_image/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ncarraz/ESRGANplus/HEAD/test_image/test.py --------------------------------------------------------------------------------