gitextract_k1iv7nsr/ ├── .gitignore ├── .pre-commit-config.yaml ├── INSTALL.md ├── LICENSE ├── MANIFEST.in ├── README.md ├── VERSION ├── basicsr/ │ ├── __init__.py │ ├── archs/ │ │ ├── BSRN_arch.py │ │ ├── Blocks.py │ │ ├── RFDN_arch.py │ │ ├── Upsamplers.py │ │ ├── __init__.py │ │ ├── arch_util.py │ │ ├── rfdnfinalB5_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 │ │ ├── paired_image_dataset.py │ │ ├── prefetch_dataloader.py │ │ ├── reds_dataset.py │ │ ├── single_image_dataset.py │ │ ├── transforms.py │ │ ├── video_test_dataset.py │ │ └── vimeo90k_dataset.py │ ├── losses/ │ │ ├── __init__.py │ │ ├── loss_util.py │ │ └── losses.py │ ├── metrics/ │ │ ├── __init__.py │ │ ├── fid.py │ │ ├── metric_util.py │ │ ├── niqe.py │ │ ├── niqe_pris_params.npz │ │ └── psnr_ssim.py │ ├── models/ │ │ ├── __init__.py │ │ ├── base_model.py │ │ ├── edvr_model.py │ │ ├── esrgan_model.py │ │ ├── hifacegan_model.py │ │ ├── lr_scheduler.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 │ ├── summary.py │ ├── test.py │ ├── train.py │ └── utils/ │ ├── __init__.py │ ├── diffjpeg.py │ ├── dist_util.py │ ├── download_util.py │ ├── face_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 ├── datasets/ │ └── 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 │ ├── api.rst │ ├── conf.py │ ├── history_updates.md │ ├── index.rst │ └── make.bat ├── experiments/ │ └── pretrained_models/ │ ├── net_g_BSRN-S.pth │ ├── net_g_BSRN_x2.pth │ ├── net_g_BSRN_x3.pth │ └── net_g_BSRN_x4.pth ├── options/ │ ├── test/ │ │ ├── benchmark_BSRN-S_x4.yml │ │ ├── benchmark_BSRN_x2.yml │ │ ├── benchmark_BSRN_x3.yml │ │ └── benchmark_BSRN_x4.yml │ └── train/ │ ├── train_BSRN-S_x4.yml │ ├── train_BSRN_x2.yml │ ├── train_BSRN_x3.yml │ └── train_BSRN_x4.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