├── README.md ├── codes ├── data │ ├── LQGT_cond_dataset.py │ ├── LQGT_dataset.py │ ├── LQ_dataset.py │ ├── README.md │ ├── REDS_dataset.py │ ├── Vimeo90K_dataset.py │ ├── __init__.py │ ├── data_sampler.py │ ├── util.py │ └── video_test_dataset.py ├── data_scripts │ ├── create_lmdb.py │ ├── extract_subimages.py │ ├── generate_2D_val.m │ ├── generate_2groups.m │ ├── generate_3D_val.m │ ├── generate_3groups.m │ ├── generate_LR_Vimeo90K.m │ ├── generate_deg.m │ ├── generate_mod_LR_bic.m │ ├── generate_mod_LR_bic.py │ ├── prepare_DIV2K_x4_dataset.sh │ ├── regroup_REDS.py │ ├── rename.py │ └── test_dataloader.py ├── metrics │ ├── calculate_PSNR_SSIM.m │ └── calculate_PSNR_SSIM.py ├── models │ ├── SRGAN_model.py │ ├── SR_model.py │ ├── Video_base_model.py │ ├── __init__.py │ ├── archs │ │ ├── AdaFMNet_arch.py │ │ ├── CResMDNet_arch.py │ │ ├── DUF_arch.py │ │ ├── EDVR_arch.py │ │ ├── RRDBNet_arch.py │ │ ├── SRResNet_arch.py │ │ ├── TOF_arch.py │ │ ├── __init__.py │ │ ├── arch_util.py │ │ ├── dcn │ │ │ ├── __init__.py │ │ │ ├── deform_conv.py │ │ │ ├── setup.py │ │ │ └── src │ │ │ │ ├── deform_conv_cuda.cpp │ │ │ │ └── deform_conv_cuda_kernel.cu │ │ └── discriminator_vgg_arch.py │ ├── base_model.py │ ├── loss.py │ ├── lr_scheduler.py │ └── networks.py ├── modulation_CResMD.py ├── options │ ├── __init__.py │ ├── options.py │ ├── test │ │ ├── modulation_CResMD.yml │ │ ├── test_AdaFMNet.yml │ │ ├── test_Base.yml │ │ ├── test_CResMD.yml │ │ ├── test_ESRGAN.yml │ │ ├── test_SRGAN.yml │ │ └── test_SRResNet.yml │ └── train │ │ ├── train_AdaFM.yml │ │ ├── train_Base.yml │ │ ├── train_CResMD.yml │ │ ├── train_EDVR_M.yml │ │ ├── train_EDVR_woTSA_M.yml │ │ ├── train_ESRGAN.yml │ │ ├── train_SRGAN.yml │ │ └── train_SRResNet.yml ├── run_scripts.sh ├── scripts │ └── transfer_params_MSRResNet.py ├── test.py ├── test_AdaFM.py ├── train.py ├── train_CResMD.py └── utils │ ├── __init__.py │ └── util.py ├── datasets └── val │ ├── blur0_noise0 │ └── zebra_mod2_0000.png │ └── blur20_noise30 │ └── zebra_mod2_2030.png ├── experiments └── pretrain_models │ ├── 2D_modulation.pth │ └── 3D_modulation.pth └── figures ├── 2D_modulation.png ├── 3D_modulation.png └── cover.png /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/README.md -------------------------------------------------------------------------------- /codes/data/LQGT_cond_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data/LQGT_cond_dataset.py -------------------------------------------------------------------------------- /codes/data/LQGT_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data/LQGT_dataset.py -------------------------------------------------------------------------------- /codes/data/LQ_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data/LQ_dataset.py -------------------------------------------------------------------------------- /codes/data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data/README.md -------------------------------------------------------------------------------- /codes/data/REDS_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data/REDS_dataset.py -------------------------------------------------------------------------------- /codes/data/Vimeo90K_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data/Vimeo90K_dataset.py -------------------------------------------------------------------------------- /codes/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data/__init__.py -------------------------------------------------------------------------------- /codes/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data/data_sampler.py -------------------------------------------------------------------------------- /codes/data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data/util.py -------------------------------------------------------------------------------- /codes/data/video_test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data/video_test_dataset.py -------------------------------------------------------------------------------- /codes/data_scripts/create_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data_scripts/create_lmdb.py -------------------------------------------------------------------------------- /codes/data_scripts/extract_subimages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data_scripts/extract_subimages.py -------------------------------------------------------------------------------- /codes/data_scripts/generate_2D_val.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data_scripts/generate_2D_val.m -------------------------------------------------------------------------------- /codes/data_scripts/generate_2groups.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data_scripts/generate_2groups.m -------------------------------------------------------------------------------- /codes/data_scripts/generate_3D_val.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data_scripts/generate_3D_val.m -------------------------------------------------------------------------------- /codes/data_scripts/generate_3groups.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data_scripts/generate_3groups.m -------------------------------------------------------------------------------- /codes/data_scripts/generate_LR_Vimeo90K.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data_scripts/generate_LR_Vimeo90K.m -------------------------------------------------------------------------------- /codes/data_scripts/generate_deg.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data_scripts/generate_deg.m -------------------------------------------------------------------------------- /codes/data_scripts/generate_mod_LR_bic.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data_scripts/generate_mod_LR_bic.m -------------------------------------------------------------------------------- /codes/data_scripts/generate_mod_LR_bic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data_scripts/generate_mod_LR_bic.py -------------------------------------------------------------------------------- /codes/data_scripts/prepare_DIV2K_x4_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data_scripts/prepare_DIV2K_x4_dataset.sh -------------------------------------------------------------------------------- /codes/data_scripts/regroup_REDS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data_scripts/regroup_REDS.py -------------------------------------------------------------------------------- /codes/data_scripts/rename.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data_scripts/rename.py -------------------------------------------------------------------------------- /codes/data_scripts/test_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/data_scripts/test_dataloader.py -------------------------------------------------------------------------------- /codes/metrics/calculate_PSNR_SSIM.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/metrics/calculate_PSNR_SSIM.m -------------------------------------------------------------------------------- /codes/metrics/calculate_PSNR_SSIM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/metrics/calculate_PSNR_SSIM.py -------------------------------------------------------------------------------- /codes/models/SRGAN_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/SRGAN_model.py -------------------------------------------------------------------------------- /codes/models/SR_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/SR_model.py -------------------------------------------------------------------------------- /codes/models/Video_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/Video_base_model.py -------------------------------------------------------------------------------- /codes/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/__init__.py -------------------------------------------------------------------------------- /codes/models/archs/AdaFMNet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/archs/AdaFMNet_arch.py -------------------------------------------------------------------------------- /codes/models/archs/CResMDNet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/archs/CResMDNet_arch.py -------------------------------------------------------------------------------- /codes/models/archs/DUF_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/archs/DUF_arch.py -------------------------------------------------------------------------------- /codes/models/archs/EDVR_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/archs/EDVR_arch.py -------------------------------------------------------------------------------- /codes/models/archs/RRDBNet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/archs/RRDBNet_arch.py -------------------------------------------------------------------------------- /codes/models/archs/SRResNet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/archs/SRResNet_arch.py -------------------------------------------------------------------------------- /codes/models/archs/TOF_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/archs/TOF_arch.py -------------------------------------------------------------------------------- /codes/models/archs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /codes/models/archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/archs/arch_util.py -------------------------------------------------------------------------------- /codes/models/archs/dcn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/archs/dcn/__init__.py -------------------------------------------------------------------------------- /codes/models/archs/dcn/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/archs/dcn/deform_conv.py -------------------------------------------------------------------------------- /codes/models/archs/dcn/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/archs/dcn/setup.py -------------------------------------------------------------------------------- /codes/models/archs/dcn/src/deform_conv_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/archs/dcn/src/deform_conv_cuda.cpp -------------------------------------------------------------------------------- /codes/models/archs/dcn/src/deform_conv_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/archs/dcn/src/deform_conv_cuda_kernel.cu -------------------------------------------------------------------------------- /codes/models/archs/discriminator_vgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/archs/discriminator_vgg_arch.py -------------------------------------------------------------------------------- /codes/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/base_model.py -------------------------------------------------------------------------------- /codes/models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/loss.py -------------------------------------------------------------------------------- /codes/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/lr_scheduler.py -------------------------------------------------------------------------------- /codes/models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/models/networks.py -------------------------------------------------------------------------------- /codes/modulation_CResMD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/modulation_CResMD.py -------------------------------------------------------------------------------- /codes/options/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /codes/options/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/options.py -------------------------------------------------------------------------------- /codes/options/test/modulation_CResMD.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/test/modulation_CResMD.yml -------------------------------------------------------------------------------- /codes/options/test/test_AdaFMNet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/test/test_AdaFMNet.yml -------------------------------------------------------------------------------- /codes/options/test/test_Base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/test/test_Base.yml -------------------------------------------------------------------------------- /codes/options/test/test_CResMD.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/test/test_CResMD.yml -------------------------------------------------------------------------------- /codes/options/test/test_ESRGAN.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/test/test_ESRGAN.yml -------------------------------------------------------------------------------- /codes/options/test/test_SRGAN.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/test/test_SRGAN.yml -------------------------------------------------------------------------------- /codes/options/test/test_SRResNet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/test/test_SRResNet.yml -------------------------------------------------------------------------------- /codes/options/train/train_AdaFM.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/train/train_AdaFM.yml -------------------------------------------------------------------------------- /codes/options/train/train_Base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/train/train_Base.yml -------------------------------------------------------------------------------- /codes/options/train/train_CResMD.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/train/train_CResMD.yml -------------------------------------------------------------------------------- /codes/options/train/train_EDVR_M.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/train/train_EDVR_M.yml -------------------------------------------------------------------------------- /codes/options/train/train_EDVR_woTSA_M.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/train/train_EDVR_woTSA_M.yml -------------------------------------------------------------------------------- /codes/options/train/train_ESRGAN.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/train/train_ESRGAN.yml -------------------------------------------------------------------------------- /codes/options/train/train_SRGAN.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/train/train_SRGAN.yml -------------------------------------------------------------------------------- /codes/options/train/train_SRResNet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/options/train/train_SRResNet.yml -------------------------------------------------------------------------------- /codes/run_scripts.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/run_scripts.sh -------------------------------------------------------------------------------- /codes/scripts/transfer_params_MSRResNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/scripts/transfer_params_MSRResNet.py -------------------------------------------------------------------------------- /codes/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/test.py -------------------------------------------------------------------------------- /codes/test_AdaFM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/test_AdaFM.py -------------------------------------------------------------------------------- /codes/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/train.py -------------------------------------------------------------------------------- /codes/train_CResMD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/train_CResMD.py -------------------------------------------------------------------------------- /codes/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /codes/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/codes/utils/util.py -------------------------------------------------------------------------------- /datasets/val/blur0_noise0/zebra_mod2_0000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/datasets/val/blur0_noise0/zebra_mod2_0000.png -------------------------------------------------------------------------------- /datasets/val/blur20_noise30/zebra_mod2_2030.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/datasets/val/blur20_noise30/zebra_mod2_2030.png -------------------------------------------------------------------------------- /experiments/pretrain_models/2D_modulation.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/experiments/pretrain_models/2D_modulation.pth -------------------------------------------------------------------------------- /experiments/pretrain_models/3D_modulation.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/experiments/pretrain_models/3D_modulation.pth -------------------------------------------------------------------------------- /figures/2D_modulation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/figures/2D_modulation.png -------------------------------------------------------------------------------- /figures/3D_modulation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/figures/3D_modulation.png -------------------------------------------------------------------------------- /figures/cover.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hejingwenhejingwen/CResMD/HEAD/figures/cover.png --------------------------------------------------------------------------------