├── .gitignore ├── CITATION.cff ├── INSTALL.md ├── LICENSE.txt ├── LICENSE ├── LICENSE ├── LICENSE-NVIDIA ├── LICENSE-stylegan2-pytorch ├── LICENSE_SwinIR ├── LICENSE_pytorch-image-models └── README.md ├── MANIFEST.in ├── README.md ├── VERSION ├── basicsr ├── __init__.py ├── archs │ ├── __init__.py │ ├── arch_util.py │ ├── basicvsr_arch.py │ ├── basicvsrpp_arch.py │ ├── contras_extractor_arch.py │ ├── contras_multi_extractor_arch.py │ ├── corres_generation_arch.py │ ├── dfdnet_arch.py │ ├── dfdnet_util.py │ ├── discriminator_arch.py │ ├── duf_arch.py │ ├── ecbsr_arch.py │ ├── edsr_arch.py │ ├── edvr_arch.py │ ├── hifacegan_arch.py │ ├── hifacegan_util.py │ ├── inception.py │ ├── rcan_arch.py │ ├── ref_map_util.py │ ├── ref_mrapa_restoration_arch.py │ ├── ref_restoration_arch.py │ ├── ridnet_arch.py │ ├── rrdbnet_arch.py │ ├── spynet_arch.py │ ├── srresnet_arch.py │ ├── srvgg_arch.py │ ├── stylegan2_arch.py │ ├── swinir_arch.py │ ├── tof_arch.py │ └── vgg_arch.py ├── data │ ├── __init__.py │ ├── data_sampler.py │ ├── data_util.py │ ├── degradations.py │ ├── ffhq_dataset.py │ ├── meta_info │ │ ├── meta_info_DIV2K800sub_GT.txt │ │ ├── meta_info_REDS4_test_GT.txt │ │ ├── meta_info_REDS_GT.txt │ │ ├── meta_info_REDSofficial4_test_GT.txt │ │ ├── meta_info_REDSval_official_test_GT.txt │ │ ├── meta_info_Vimeo90K_test_GT.txt │ │ ├── meta_info_Vimeo90K_test_fast_GT.txt │ │ ├── meta_info_Vimeo90K_test_medium_GT.txt │ │ ├── meta_info_Vimeo90K_test_slow_GT.txt │ │ └── meta_info_Vimeo90K_train_GT.txt │ ├── multi_ref_dataset.py │ ├── paired_image_dataset.py │ ├── prefetch_dataloader.py │ ├── realesrgan_dataset.py │ ├── realesrgan_paired_dataset.py │ ├── reds_dataset.py │ ├── ref_cufed_dataset.py │ ├── ref_megadepth_dataset.py │ ├── single_image_dataset.py │ ├── single_ref_dataset.py │ ├── sisr_dataset.py │ ├── transforms.py │ ├── video_test_dataset.py │ └── vimeo90k_dataset.py ├── losses │ ├── __init__.py │ ├── loss_util.py │ └── losses.py ├── metrics │ ├── README.md │ ├── README_CN.md │ ├── __init__.py │ ├── fid.py │ ├── metric_util.py │ ├── niqe.py │ ├── niqe_pris_params.npz │ ├── psnr_ssim.py │ └── test_metrics │ │ └── test_psnr_ssim.py ├── models │ ├── __init__.py │ ├── base_model.py │ ├── edvr_model.py │ ├── esrgan_model.py │ ├── hifacegan_model.py │ ├── loss_utils.py │ ├── losses.py │ ├── lr_scheduler.py │ ├── multi_ref_restoration_model.py │ ├── realesrgan_model.py │ ├── realesrnet_model.py │ ├── ref_restoration_model.py │ ├── sr_model.py │ ├── srgan_model.py │ ├── stylegan2_model.py │ ├── swinir_model.py │ ├── video_base_model.py │ ├── video_gan_model.py │ ├── video_recurrent_gan_model.py │ └── video_recurrent_model.py ├── ops │ ├── __init__.py │ ├── dcn │ │ ├── __init__.py │ │ ├── deform_conv.py │ │ └── src │ │ │ ├── deform_conv_cuda.cpp │ │ │ ├── deform_conv_cuda_kernel.cu │ │ │ └── deform_conv_ext.cpp │ ├── fused_act │ │ ├── __init__.py │ │ ├── fused_act.py │ │ └── src │ │ │ ├── fused_bias_act.cpp │ │ │ └── fused_bias_act_kernel.cu │ └── upfirdn2d │ │ ├── __init__.py │ │ ├── src │ │ ├── upfirdn2d.cpp │ │ └── upfirdn2d_kernel.cu │ │ └── upfirdn2d.py ├── test.py ├── train.py └── utils │ ├── __init__.py │ ├── color_util.py │ ├── diffjpeg.py │ ├── dist_util.py │ ├── download_util.py │ ├── file_client.py │ ├── flow_util.py │ ├── img_process_util.py │ ├── img_util.py │ ├── lmdb_util.py │ ├── logger.py │ ├── matlab_functions.py │ ├── misc.py │ ├── options.py │ └── registry.py ├── colab └── README.md ├── docs ├── Config.md ├── Config_CN.md ├── DatasetPreparation.md ├── DatasetPreparation_CN.md ├── Datasets.md ├── Datasets_CN.md ├── DesignConvention.md ├── DesignConvention_CN.md ├── HOWTOs.md ├── HOWTOs_CN.md ├── Logging.md ├── Logging_CN.md ├── Makefile ├── Metrics.md ├── Metrics_CN.md ├── ModelZoo.md ├── ModelZoo_CN.md ├── Models.md ├── Models_CN.md ├── TrainTest.md ├── TrainTest_CN.md ├── auto_generate_api.py ├── conf.py ├── history_updates.md ├── index.rst ├── make.bat └── requirements.txt ├── inference ├── inference_basicvsr.py ├── inference_basicvsrpp.py ├── inference_dfdnet.py ├── inference_esrgan.py ├── inference_ridnet.py ├── inference_stylegan2.py └── inference_swinir.py ├── options └── train │ ├── stage3_5ref_restoration_mse.yml │ └── stage3_5ref_restoration_mse_lp.yml ├── requirements.txt ├── scripts ├── data_preparation │ ├── create_lmdb.py │ ├── download_datasets.py │ ├── extract_images_from_tfrecords.py │ ├── extract_subimages.py │ ├── generate_meta_info.py │ ├── prepare_hifacegan_dataset.py │ └── regroup_reds_dataset.py ├── dist_test.sh ├── dist_train.sh ├── download_gdrive.py ├── download_pretrained_models.py ├── matlab_scripts │ ├── back_projection │ │ ├── backprojection.m │ │ ├── main_bp.m │ │ └── main_reverse_filter.m │ ├── generate_LR_Vimeo90K.m │ └── generate_bicubic_img.m ├── metrics │ ├── calculate_fid_folder.py │ ├── calculate_fid_stats_from_datasets.py │ ├── calculate_lpips.py │ ├── calculate_niqe.py │ ├── calculate_psnr_ssim.py │ └── calculate_stylegan2_fid.py ├── model_conversion │ ├── convert_dfdnet.py │ ├── convert_models.py │ ├── convert_ridnet.py │ └── convert_stylegan.py └── publish_models.py ├── setup.cfg ├── setup.py └── tests ├── README.md ├── data ├── gt.lmdb │ ├── data.mdb │ ├── lock.mdb │ └── meta_info.txt ├── lq.lmdb │ ├── data.mdb │ ├── lock.mdb │ └── meta_info.txt ├── meta_info_gt.txt └── meta_info_pair.txt ├── test_archs ├── test_basicvsr_arch.py ├── test_discriminator_arch.py ├── test_duf_arch.py ├── test_ecbsr_arch.py └── test_srresnet_arch.py ├── test_data ├── test_paired_image_dataset.py └── test_single_image_dataset.py ├── test_losses └── test_losses.py ├── test_metrics └── test_psnr_ssim.py └── test_models └── test_sr_model.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/CITATION.cff -------------------------------------------------------------------------------- /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/INSTALL.md -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /LICENSE/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/LICENSE/LICENSE -------------------------------------------------------------------------------- /LICENSE/LICENSE-NVIDIA: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/LICENSE/LICENSE-NVIDIA -------------------------------------------------------------------------------- /LICENSE/LICENSE-stylegan2-pytorch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/LICENSE/LICENSE-stylegan2-pytorch -------------------------------------------------------------------------------- /LICENSE/LICENSE_SwinIR: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/LICENSE/LICENSE_SwinIR -------------------------------------------------------------------------------- /LICENSE/LICENSE_pytorch-image-models: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/LICENSE/LICENSE_pytorch-image-models -------------------------------------------------------------------------------- /LICENSE/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/LICENSE/README.md -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/README.md -------------------------------------------------------------------------------- /VERSION: -------------------------------------------------------------------------------- 1 | 1.3.5 2 | -------------------------------------------------------------------------------- /basicsr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/arch_util.py -------------------------------------------------------------------------------- /basicsr/archs/basicvsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/basicvsr_arch.py -------------------------------------------------------------------------------- /basicsr/archs/basicvsrpp_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/basicvsrpp_arch.py -------------------------------------------------------------------------------- /basicsr/archs/contras_extractor_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/contras_extractor_arch.py -------------------------------------------------------------------------------- /basicsr/archs/contras_multi_extractor_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/contras_multi_extractor_arch.py -------------------------------------------------------------------------------- /basicsr/archs/corres_generation_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/corres_generation_arch.py -------------------------------------------------------------------------------- /basicsr/archs/dfdnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/dfdnet_arch.py -------------------------------------------------------------------------------- /basicsr/archs/dfdnet_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/dfdnet_util.py -------------------------------------------------------------------------------- /basicsr/archs/discriminator_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/discriminator_arch.py -------------------------------------------------------------------------------- /basicsr/archs/duf_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/duf_arch.py -------------------------------------------------------------------------------- /basicsr/archs/ecbsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/ecbsr_arch.py -------------------------------------------------------------------------------- /basicsr/archs/edsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/edsr_arch.py -------------------------------------------------------------------------------- /basicsr/archs/edvr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/edvr_arch.py -------------------------------------------------------------------------------- /basicsr/archs/hifacegan_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/hifacegan_arch.py -------------------------------------------------------------------------------- /basicsr/archs/hifacegan_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/hifacegan_util.py -------------------------------------------------------------------------------- /basicsr/archs/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/inception.py -------------------------------------------------------------------------------- /basicsr/archs/rcan_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/rcan_arch.py -------------------------------------------------------------------------------- /basicsr/archs/ref_map_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/ref_map_util.py -------------------------------------------------------------------------------- /basicsr/archs/ref_mrapa_restoration_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/ref_mrapa_restoration_arch.py -------------------------------------------------------------------------------- /basicsr/archs/ref_restoration_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/ref_restoration_arch.py -------------------------------------------------------------------------------- /basicsr/archs/ridnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/ridnet_arch.py -------------------------------------------------------------------------------- /basicsr/archs/rrdbnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/rrdbnet_arch.py -------------------------------------------------------------------------------- /basicsr/archs/spynet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/spynet_arch.py -------------------------------------------------------------------------------- /basicsr/archs/srresnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/srresnet_arch.py -------------------------------------------------------------------------------- /basicsr/archs/srvgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/srvgg_arch.py -------------------------------------------------------------------------------- /basicsr/archs/stylegan2_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/stylegan2_arch.py -------------------------------------------------------------------------------- /basicsr/archs/swinir_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/swinir_arch.py -------------------------------------------------------------------------------- /basicsr/archs/tof_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/tof_arch.py -------------------------------------------------------------------------------- /basicsr/archs/vgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/archs/vgg_arch.py -------------------------------------------------------------------------------- /basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/__init__.py -------------------------------------------------------------------------------- /basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/data_util.py -------------------------------------------------------------------------------- /basicsr/data/degradations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/degradations.py -------------------------------------------------------------------------------- /basicsr/data/ffhq_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/ffhq_dataset.py -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_DIV2K800sub_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/meta_info/meta_info_DIV2K800sub_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDS4_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/meta_info/meta_info_REDS4_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDS_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/meta_info/meta_info_REDS_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDSofficial4_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/meta_info/meta_info_REDSofficial4_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDSval_official_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/meta_info/meta_info_REDSval_official_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_test_fast_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_test_fast_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_test_medium_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_test_medium_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_test_slow_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_test_slow_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_train_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_train_GT.txt -------------------------------------------------------------------------------- /basicsr/data/multi_ref_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/multi_ref_dataset.py -------------------------------------------------------------------------------- /basicsr/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/paired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /basicsr/data/realesrgan_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/realesrgan_dataset.py -------------------------------------------------------------------------------- /basicsr/data/realesrgan_paired_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/realesrgan_paired_dataset.py -------------------------------------------------------------------------------- /basicsr/data/reds_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/reds_dataset.py -------------------------------------------------------------------------------- /basicsr/data/ref_cufed_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/ref_cufed_dataset.py -------------------------------------------------------------------------------- /basicsr/data/ref_megadepth_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/ref_megadepth_dataset.py -------------------------------------------------------------------------------- /basicsr/data/single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/single_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/single_ref_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/single_ref_dataset.py -------------------------------------------------------------------------------- /basicsr/data/sisr_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/sisr_dataset.py -------------------------------------------------------------------------------- /basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/transforms.py -------------------------------------------------------------------------------- /basicsr/data/video_test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/video_test_dataset.py -------------------------------------------------------------------------------- /basicsr/data/vimeo90k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/data/vimeo90k_dataset.py -------------------------------------------------------------------------------- /basicsr/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/losses/__init__.py -------------------------------------------------------------------------------- /basicsr/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/losses/loss_util.py -------------------------------------------------------------------------------- /basicsr/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/losses/losses.py -------------------------------------------------------------------------------- /basicsr/metrics/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/metrics/README.md -------------------------------------------------------------------------------- /basicsr/metrics/README_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/metrics/README_CN.md -------------------------------------------------------------------------------- /basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /basicsr/metrics/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/metrics/fid.py -------------------------------------------------------------------------------- /basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/metrics/niqe.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe_pris_params.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/metrics/niqe_pris_params.npz -------------------------------------------------------------------------------- /basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/metrics/test_metrics/test_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/metrics/test_metrics/test_psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/__init__.py -------------------------------------------------------------------------------- /basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/base_model.py -------------------------------------------------------------------------------- /basicsr/models/edvr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/edvr_model.py -------------------------------------------------------------------------------- /basicsr/models/esrgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/esrgan_model.py -------------------------------------------------------------------------------- /basicsr/models/hifacegan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/hifacegan_model.py -------------------------------------------------------------------------------- /basicsr/models/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/loss_utils.py -------------------------------------------------------------------------------- /basicsr/models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/losses.py -------------------------------------------------------------------------------- /basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /basicsr/models/multi_ref_restoration_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/multi_ref_restoration_model.py -------------------------------------------------------------------------------- /basicsr/models/realesrgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/realesrgan_model.py -------------------------------------------------------------------------------- /basicsr/models/realesrnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/realesrnet_model.py -------------------------------------------------------------------------------- /basicsr/models/ref_restoration_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/ref_restoration_model.py -------------------------------------------------------------------------------- /basicsr/models/sr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/sr_model.py -------------------------------------------------------------------------------- /basicsr/models/srgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/srgan_model.py -------------------------------------------------------------------------------- /basicsr/models/stylegan2_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/stylegan2_model.py -------------------------------------------------------------------------------- /basicsr/models/swinir_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/swinir_model.py -------------------------------------------------------------------------------- /basicsr/models/video_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/video_base_model.py -------------------------------------------------------------------------------- /basicsr/models/video_gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/video_gan_model.py -------------------------------------------------------------------------------- /basicsr/models/video_recurrent_gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/video_recurrent_gan_model.py -------------------------------------------------------------------------------- /basicsr/models/video_recurrent_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/models/video_recurrent_model.py -------------------------------------------------------------------------------- /basicsr/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /basicsr/ops/dcn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/ops/dcn/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/dcn/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/ops/dcn/deform_conv.py -------------------------------------------------------------------------------- /basicsr/ops/dcn/src/deform_conv_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/ops/dcn/src/deform_conv_cuda.cpp -------------------------------------------------------------------------------- /basicsr/ops/dcn/src/deform_conv_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/ops/dcn/src/deform_conv_cuda_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/dcn/src/deform_conv_ext.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/ops/dcn/src/deform_conv_ext.cpp -------------------------------------------------------------------------------- /basicsr/ops/fused_act/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/ops/fused_act/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/fused_act/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/ops/fused_act/fused_act.py -------------------------------------------------------------------------------- /basicsr/ops/fused_act/src/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/ops/fused_act/src/fused_bias_act.cpp -------------------------------------------------------------------------------- /basicsr/ops/fused_act/src/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/ops/fused_act/src/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/ops/upfirdn2d/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/src/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/ops/upfirdn2d/src/upfirdn2d.cpp -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/src/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/ops/upfirdn2d/src/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/ops/upfirdn2d/upfirdn2d.py -------------------------------------------------------------------------------- /basicsr/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/test.py -------------------------------------------------------------------------------- /basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/train.py -------------------------------------------------------------------------------- /basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/color_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/color_util.py -------------------------------------------------------------------------------- /basicsr/utils/diffjpeg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/diffjpeg.py -------------------------------------------------------------------------------- /basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /basicsr/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/download_util.py -------------------------------------------------------------------------------- /basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/file_client.py -------------------------------------------------------------------------------- /basicsr/utils/flow_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/flow_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_process_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/img_process_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/img_util.py -------------------------------------------------------------------------------- /basicsr/utils/lmdb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/lmdb_util.py -------------------------------------------------------------------------------- /basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/logger.py -------------------------------------------------------------------------------- /basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/misc.py -------------------------------------------------------------------------------- /basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/options.py -------------------------------------------------------------------------------- /basicsr/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/basicsr/utils/registry.py -------------------------------------------------------------------------------- /colab/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/colab/README.md -------------------------------------------------------------------------------- /docs/Config.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/Config.md -------------------------------------------------------------------------------- /docs/Config_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/Config_CN.md -------------------------------------------------------------------------------- /docs/DatasetPreparation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/DatasetPreparation.md -------------------------------------------------------------------------------- /docs/DatasetPreparation_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/DatasetPreparation_CN.md -------------------------------------------------------------------------------- /docs/Datasets.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/Datasets.md -------------------------------------------------------------------------------- /docs/Datasets_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/Datasets_CN.md -------------------------------------------------------------------------------- /docs/DesignConvention.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/DesignConvention.md -------------------------------------------------------------------------------- /docs/DesignConvention_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/DesignConvention_CN.md -------------------------------------------------------------------------------- /docs/HOWTOs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/HOWTOs.md -------------------------------------------------------------------------------- /docs/HOWTOs_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/HOWTOs_CN.md -------------------------------------------------------------------------------- /docs/Logging.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/Logging.md -------------------------------------------------------------------------------- /docs/Logging_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/Logging_CN.md -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/Metrics.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/Metrics.md -------------------------------------------------------------------------------- /docs/Metrics_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/Metrics_CN.md -------------------------------------------------------------------------------- /docs/ModelZoo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/ModelZoo.md -------------------------------------------------------------------------------- /docs/ModelZoo_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/ModelZoo_CN.md -------------------------------------------------------------------------------- /docs/Models.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/Models.md -------------------------------------------------------------------------------- /docs/Models_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/Models_CN.md -------------------------------------------------------------------------------- /docs/TrainTest.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/TrainTest.md -------------------------------------------------------------------------------- /docs/TrainTest_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/TrainTest_CN.md -------------------------------------------------------------------------------- /docs/auto_generate_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/auto_generate_api.py -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/history_updates.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/history_updates.md -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /inference/inference_basicvsr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/inference/inference_basicvsr.py -------------------------------------------------------------------------------- /inference/inference_basicvsrpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/inference/inference_basicvsrpp.py -------------------------------------------------------------------------------- /inference/inference_dfdnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/inference/inference_dfdnet.py -------------------------------------------------------------------------------- /inference/inference_esrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/inference/inference_esrgan.py -------------------------------------------------------------------------------- /inference/inference_ridnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/inference/inference_ridnet.py -------------------------------------------------------------------------------- /inference/inference_stylegan2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/inference/inference_stylegan2.py -------------------------------------------------------------------------------- /inference/inference_swinir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/inference/inference_swinir.py -------------------------------------------------------------------------------- /options/train/stage3_5ref_restoration_mse.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/options/train/stage3_5ref_restoration_mse.yml -------------------------------------------------------------------------------- /options/train/stage3_5ref_restoration_mse_lp.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/options/train/stage3_5ref_restoration_mse_lp.yml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/data_preparation/create_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/data_preparation/create_lmdb.py -------------------------------------------------------------------------------- /scripts/data_preparation/download_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/data_preparation/download_datasets.py -------------------------------------------------------------------------------- /scripts/data_preparation/extract_images_from_tfrecords.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/data_preparation/extract_images_from_tfrecords.py -------------------------------------------------------------------------------- /scripts/data_preparation/extract_subimages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/data_preparation/extract_subimages.py -------------------------------------------------------------------------------- /scripts/data_preparation/generate_meta_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/data_preparation/generate_meta_info.py -------------------------------------------------------------------------------- /scripts/data_preparation/prepare_hifacegan_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/data_preparation/prepare_hifacegan_dataset.py -------------------------------------------------------------------------------- /scripts/data_preparation/regroup_reds_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/data_preparation/regroup_reds_dataset.py -------------------------------------------------------------------------------- /scripts/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/dist_test.sh -------------------------------------------------------------------------------- /scripts/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/dist_train.sh -------------------------------------------------------------------------------- /scripts/download_gdrive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/download_gdrive.py -------------------------------------------------------------------------------- /scripts/download_pretrained_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/download_pretrained_models.py -------------------------------------------------------------------------------- /scripts/matlab_scripts/back_projection/backprojection.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/matlab_scripts/back_projection/backprojection.m -------------------------------------------------------------------------------- /scripts/matlab_scripts/back_projection/main_bp.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/matlab_scripts/back_projection/main_bp.m -------------------------------------------------------------------------------- /scripts/matlab_scripts/back_projection/main_reverse_filter.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/matlab_scripts/back_projection/main_reverse_filter.m -------------------------------------------------------------------------------- /scripts/matlab_scripts/generate_LR_Vimeo90K.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/matlab_scripts/generate_LR_Vimeo90K.m -------------------------------------------------------------------------------- /scripts/matlab_scripts/generate_bicubic_img.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/matlab_scripts/generate_bicubic_img.m -------------------------------------------------------------------------------- /scripts/metrics/calculate_fid_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/metrics/calculate_fid_folder.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_fid_stats_from_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/metrics/calculate_fid_stats_from_datasets.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/metrics/calculate_lpips.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/metrics/calculate_niqe.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/metrics/calculate_psnr_ssim.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_stylegan2_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/metrics/calculate_stylegan2_fid.py -------------------------------------------------------------------------------- /scripts/model_conversion/convert_dfdnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/model_conversion/convert_dfdnet.py -------------------------------------------------------------------------------- /scripts/model_conversion/convert_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/model_conversion/convert_models.py -------------------------------------------------------------------------------- /scripts/model_conversion/convert_ridnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/model_conversion/convert_ridnet.py -------------------------------------------------------------------------------- /scripts/model_conversion/convert_stylegan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/model_conversion/convert_stylegan.py -------------------------------------------------------------------------------- /scripts/publish_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/scripts/publish_models.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/setup.py -------------------------------------------------------------------------------- /tests/README.md: -------------------------------------------------------------------------------- 1 | # UnitTest 2 | 3 | - It requires GPU CUDA environment 4 | -------------------------------------------------------------------------------- /tests/data/gt.lmdb/data.mdb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/data/gt.lmdb/data.mdb -------------------------------------------------------------------------------- /tests/data/gt.lmdb/lock.mdb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/data/gt.lmdb/lock.mdb -------------------------------------------------------------------------------- /tests/data/gt.lmdb/meta_info.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/data/gt.lmdb/meta_info.txt -------------------------------------------------------------------------------- /tests/data/lq.lmdb/data.mdb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/data/lq.lmdb/data.mdb -------------------------------------------------------------------------------- /tests/data/lq.lmdb/lock.mdb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/data/lq.lmdb/lock.mdb -------------------------------------------------------------------------------- /tests/data/lq.lmdb/meta_info.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/data/lq.lmdb/meta_info.txt -------------------------------------------------------------------------------- /tests/data/meta_info_gt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/data/meta_info_gt.txt -------------------------------------------------------------------------------- /tests/data/meta_info_pair.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/data/meta_info_pair.txt -------------------------------------------------------------------------------- /tests/test_archs/test_basicvsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/test_archs/test_basicvsr_arch.py -------------------------------------------------------------------------------- /tests/test_archs/test_discriminator_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/test_archs/test_discriminator_arch.py -------------------------------------------------------------------------------- /tests/test_archs/test_duf_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/test_archs/test_duf_arch.py -------------------------------------------------------------------------------- /tests/test_archs/test_ecbsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/test_archs/test_ecbsr_arch.py -------------------------------------------------------------------------------- /tests/test_archs/test_srresnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/test_archs/test_srresnet_arch.py -------------------------------------------------------------------------------- /tests/test_data/test_paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/test_data/test_paired_image_dataset.py -------------------------------------------------------------------------------- /tests/test_data/test_single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/test_data/test_single_image_dataset.py -------------------------------------------------------------------------------- /tests/test_losses/test_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/test_losses/test_losses.py -------------------------------------------------------------------------------- /tests/test_metrics/test_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/test_metrics/test_psnr_ssim.py -------------------------------------------------------------------------------- /tests/test_models/test_sr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wdmwhh/MRefSR/HEAD/tests/test_models/test_sr_model.py --------------------------------------------------------------------------------