├── .gitignore ├── .pre-commit-config.yaml ├── .readthedocs.yaml ├── CITATION.cff ├── LICENSE.txt ├── MANIFEST.in ├── README.md ├── VERSION ├── basicsr ├── __init__.py ├── archs │ ├── ECBVSRUni_arch.py │ ├── ECBVSR_arch.py │ ├── __init__.py │ ├── arch_util.py │ ├── basicvsr_arch.py │ ├── ecbsr_arch.py │ ├── edsr_arch.py │ ├── edvr_arch.py │ ├── rcan_arch.py │ ├── spynet_arch.py │ ├── swinir_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 │ ├── realesrgan_dataset.py │ ├── realesrgan_paired_dataset.py │ ├── reds_dataset.py │ ├── single_image_dataset.py │ ├── transforms.py │ ├── video_test_dataset.py │ └── vimeo90k_dataset.py ├── losses │ ├── __init__.py │ ├── basic_loss.py │ ├── gan_loss.py │ └── loss_util.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 │ ├── lr_scheduler.py │ ├── realesrgan_model.py │ ├── realesrnet_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 │ ├── plot_util.py │ └── registry.py ├── inference ├── inference_basicvsr.py ├── inference_basicvsrpp.py ├── inference_dfdnet.py ├── inference_esrgan.py ├── inference_ridnet.py ├── inference_stylegan2.py └── inference_swinir.py ├── options ├── test │ ├── BasicVSR │ │ ├── test_BasicVSR_REDS.yml │ │ ├── test_BasicVSR_Vimeo90K_BDx4.yml │ │ ├── test_BasicVSR_Vimeo90K_BIx4.yml │ │ ├── test_IconVSR_REDS.yml │ │ ├── test_IconVSR_Vimeo90K_BDx4.yml │ │ └── test_IconVSR_Vimeo90K_BIx4.yml │ ├── DUF │ │ └── test_DUF_official.yml │ ├── ECBVSR │ │ ├── test_ECBVSR_REDS.yml │ │ └── test_ECBVSR_Vimeo.yml │ ├── EDSR │ │ ├── test_EDSR_Lx2.yml │ │ ├── test_EDSR_Lx3.yml │ │ ├── test_EDSR_Lx4.yml │ │ ├── test_EDSR_Mx2.yml │ │ ├── test_EDSR_Mx3.yml │ │ └── test_EDSR_Mx4.yml │ ├── EDVR │ │ ├── test_EDVR_L_deblur_REDS.yml │ │ ├── test_EDVR_L_deblurcomp_REDS.yml │ │ ├── test_EDVR_L_x4_SR_REDS.yml │ │ ├── test_EDVR_L_x4_SR_Vid4.yml │ │ ├── test_EDVR_L_x4_SR_Vimeo90K.yml │ │ ├── test_EDVR_L_x4_SRblur_REDS.yml │ │ └── test_EDVR_M_x4_SR_REDS.yml │ ├── ESRGAN │ │ ├── test_ESRGAN_x4.yml │ │ ├── test_ESRGAN_x4_woGT.yml │ │ └── test_RRDBNet_PSNR_x4.yml │ ├── HiFaceGAN │ │ ├── test_hifacegan.yml │ │ └── test_hifacegan_woGT.yml │ ├── RCAN │ │ └── test_RCAN.yml │ ├── SRResNet_SRGAN │ │ ├── test_MSRGAN_x4.yml │ │ ├── test_MSRResNet_x2.yml │ │ ├── test_MSRResNet_x3.yml │ │ ├── test_MSRResNet_x4.yml │ │ └── test_MSRResNet_x4_woGT.yml │ └── TOF │ │ └── test_TOF_official.yml └── train │ ├── BasicVSR │ ├── train_BasicVSR_REDS.yml │ ├── train_BasicVSR_Vimeo90K_BDx4.yml │ ├── train_BasicVSR_Vimeo90K_BIx4.yml │ ├── train_IconVSR_REDS.yml │ ├── train_IconVSR_Vimeo90K_BDx4.yml │ └── train_IconVSR_Vimeo90K_BIx4.yml │ ├── BasicVSRPP │ └── train_BasicVSRPP_REDS.yml │ ├── ECBSR │ ├── train_ECBSR_x2_m4c16_prelu.yml │ ├── train_ECBSR_x4_m4c16_prelu.yml │ └── train_ECBSR_x4_m4c16_prelu_RGB.yml │ ├── ECBVSR │ ├── train_ECBVSR_REDS.yml │ └── train_ECBVSR_Vimeo.yml │ ├── EDSR │ ├── train_EDSR_Lx2.yml │ ├── train_EDSR_Lx3.yml │ ├── train_EDSR_Lx4.yml │ ├── train_EDSR_Mx2.yml │ ├── train_EDSR_Mx3.yml │ └── train_EDSR_Mx4.yml │ ├── EDVR │ ├── train_EDVRM_woTSA_GAN_TODO.yml │ ├── train_EDVR_L_x4_SR_REDS.yml │ ├── train_EDVR_L_x4_SR_REDS_woTSA.yml │ ├── train_EDVR_M_x4_SR_REDS.yml │ └── train_EDVR_M_x4_SR_REDS_woTSA.yml │ ├── ESRGAN │ ├── train_ESRGAN_x4.yml │ └── train_RRDBNet_PSNR_x4.yml │ ├── HiFaceGAN │ └── train_hifacegan.yml │ ├── LDL │ └── train_LDL_Real_x4.yml │ ├── RCAN │ └── train_RCAN_x2.yml │ ├── RealESRGAN │ ├── train_realesrgan_x2plus.yml │ ├── train_realesrgan_x4plus.yml │ ├── train_realesrnet_x2plus.yml │ └── train_realesrnet_x4plus.yml │ ├── SRResNet_SRGAN │ ├── README.md │ ├── train_MSRGAN_x4.yml │ ├── train_MSRResNet_x2.yml │ ├── train_MSRResNet_x3.yml │ └── train_MSRResNet_x4.yml │ ├── StyleGAN │ └── train_StyleGAN2_256_Cmul2_FFHQ.yml │ ├── SwinIR │ ├── train_SwinIR_SRx2_scratch.yml │ └── train_SwinIR_SRx4_scratch.yml │ └── VideoRecurrentGAN │ └── train_VideoRecurrentGANModel_REDS.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 ├── plot │ ├── README.md │ └── model_complexity_cmp_bsrn.py └── publish_models.py ├── setup.cfg ├── setup.py ├── test_scripts ├── test_discriminator_backward.py ├── test_ffhq_dataset.py ├── test_lr_scheduler.py ├── test_niqe.py ├── test_paired_image_dataset.py ├── test_reds_dataset.py └── test_vimeo90k_dataset.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/Kimsure/KSNet/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/README.md -------------------------------------------------------------------------------- /VERSION: -------------------------------------------------------------------------------- 1 | 1.4.2 2 | -------------------------------------------------------------------------------- /basicsr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/ECBVSRUni_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/archs/ECBVSRUni_arch.py -------------------------------------------------------------------------------- /basicsr/archs/ECBVSR_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/archs/ECBVSR_arch.py -------------------------------------------------------------------------------- /basicsr/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/archs/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/archs/arch_util.py -------------------------------------------------------------------------------- /basicsr/archs/basicvsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/archs/basicvsr_arch.py -------------------------------------------------------------------------------- /basicsr/archs/ecbsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/archs/ecbsr_arch.py -------------------------------------------------------------------------------- /basicsr/archs/edsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/archs/edsr_arch.py -------------------------------------------------------------------------------- /basicsr/archs/edvr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/archs/edvr_arch.py -------------------------------------------------------------------------------- /basicsr/archs/rcan_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/archs/rcan_arch.py -------------------------------------------------------------------------------- /basicsr/archs/spynet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/archs/spynet_arch.py -------------------------------------------------------------------------------- /basicsr/archs/swinir_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/archs/swinir_arch.py -------------------------------------------------------------------------------- /basicsr/archs/vgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/archs/vgg_arch.py -------------------------------------------------------------------------------- /basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/__init__.py -------------------------------------------------------------------------------- /basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/data_util.py -------------------------------------------------------------------------------- /basicsr/data/degradations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/degradations.py -------------------------------------------------------------------------------- /basicsr/data/ffhq_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/ffhq_dataset.py -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_DIV2K800sub_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/meta_info/meta_info_DIV2K800sub_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDS4_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/meta_info/meta_info_REDS4_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDS_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/meta_info/meta_info_REDS_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDSofficial4_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/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/Kimsure/KSNet/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/Kimsure/KSNet/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/Kimsure/KSNet/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/Kimsure/KSNet/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/Kimsure/KSNet/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/Kimsure/KSNet/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_train_GT.txt -------------------------------------------------------------------------------- /basicsr/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/paired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /basicsr/data/realesrgan_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/realesrgan_dataset.py -------------------------------------------------------------------------------- /basicsr/data/realesrgan_paired_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/realesrgan_paired_dataset.py -------------------------------------------------------------------------------- /basicsr/data/reds_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/reds_dataset.py -------------------------------------------------------------------------------- /basicsr/data/single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/single_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/transforms.py -------------------------------------------------------------------------------- /basicsr/data/video_test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/video_test_dataset.py -------------------------------------------------------------------------------- /basicsr/data/vimeo90k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/data/vimeo90k_dataset.py -------------------------------------------------------------------------------- /basicsr/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/losses/__init__.py -------------------------------------------------------------------------------- /basicsr/losses/basic_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/losses/basic_loss.py -------------------------------------------------------------------------------- /basicsr/losses/gan_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/losses/gan_loss.py -------------------------------------------------------------------------------- /basicsr/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/losses/loss_util.py -------------------------------------------------------------------------------- /basicsr/metrics/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/metrics/README.md -------------------------------------------------------------------------------- /basicsr/metrics/README_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/metrics/README_CN.md -------------------------------------------------------------------------------- /basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /basicsr/metrics/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/metrics/fid.py -------------------------------------------------------------------------------- /basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/metrics/niqe.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe_pris_params.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/metrics/niqe_pris_params.npz -------------------------------------------------------------------------------- /basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/metrics/test_metrics/test_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/metrics/test_metrics/test_psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/__init__.py -------------------------------------------------------------------------------- /basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/base_model.py -------------------------------------------------------------------------------- /basicsr/models/edvr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/edvr_model.py -------------------------------------------------------------------------------- /basicsr/models/esrgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/esrgan_model.py -------------------------------------------------------------------------------- /basicsr/models/hifacegan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/hifacegan_model.py -------------------------------------------------------------------------------- /basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /basicsr/models/realesrgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/realesrgan_model.py -------------------------------------------------------------------------------- /basicsr/models/realesrnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/realesrnet_model.py -------------------------------------------------------------------------------- /basicsr/models/sr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/sr_model.py -------------------------------------------------------------------------------- /basicsr/models/srgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/srgan_model.py -------------------------------------------------------------------------------- /basicsr/models/stylegan2_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/stylegan2_model.py -------------------------------------------------------------------------------- /basicsr/models/swinir_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/swinir_model.py -------------------------------------------------------------------------------- /basicsr/models/video_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/video_base_model.py -------------------------------------------------------------------------------- /basicsr/models/video_gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/video_gan_model.py -------------------------------------------------------------------------------- /basicsr/models/video_recurrent_gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/video_recurrent_gan_model.py -------------------------------------------------------------------------------- /basicsr/models/video_recurrent_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/models/video_recurrent_model.py -------------------------------------------------------------------------------- /basicsr/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /basicsr/ops/dcn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/ops/dcn/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/dcn/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/ops/dcn/deform_conv.py -------------------------------------------------------------------------------- /basicsr/ops/dcn/src/deform_conv_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/ops/dcn/src/deform_conv_cuda.cpp -------------------------------------------------------------------------------- /basicsr/ops/dcn/src/deform_conv_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/ops/dcn/src/deform_conv_cuda_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/dcn/src/deform_conv_ext.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/ops/dcn/src/deform_conv_ext.cpp -------------------------------------------------------------------------------- /basicsr/ops/fused_act/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/ops/fused_act/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/fused_act/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/ops/fused_act/fused_act.py -------------------------------------------------------------------------------- /basicsr/ops/fused_act/src/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/ops/fused_act/src/fused_bias_act.cpp -------------------------------------------------------------------------------- /basicsr/ops/fused_act/src/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/ops/fused_act/src/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/ops/upfirdn2d/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/src/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/ops/upfirdn2d/src/upfirdn2d.cpp -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/src/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/ops/upfirdn2d/src/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/ops/upfirdn2d/upfirdn2d.py -------------------------------------------------------------------------------- /basicsr/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/test.py -------------------------------------------------------------------------------- /basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/train.py -------------------------------------------------------------------------------- /basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/color_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/color_util.py -------------------------------------------------------------------------------- /basicsr/utils/diffjpeg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/diffjpeg.py -------------------------------------------------------------------------------- /basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /basicsr/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/download_util.py -------------------------------------------------------------------------------- /basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/file_client.py -------------------------------------------------------------------------------- /basicsr/utils/flow_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/flow_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_process_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/img_process_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/img_util.py -------------------------------------------------------------------------------- /basicsr/utils/lmdb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/lmdb_util.py -------------------------------------------------------------------------------- /basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/logger.py -------------------------------------------------------------------------------- /basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/misc.py -------------------------------------------------------------------------------- /basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/options.py -------------------------------------------------------------------------------- /basicsr/utils/plot_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/plot_util.py -------------------------------------------------------------------------------- /basicsr/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/basicsr/utils/registry.py -------------------------------------------------------------------------------- /inference/inference_basicvsr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/inference/inference_basicvsr.py -------------------------------------------------------------------------------- /inference/inference_basicvsrpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/inference/inference_basicvsrpp.py -------------------------------------------------------------------------------- /inference/inference_dfdnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/inference/inference_dfdnet.py -------------------------------------------------------------------------------- /inference/inference_esrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/inference/inference_esrgan.py -------------------------------------------------------------------------------- /inference/inference_ridnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/inference/inference_ridnet.py -------------------------------------------------------------------------------- /inference/inference_stylegan2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/inference/inference_stylegan2.py -------------------------------------------------------------------------------- /inference/inference_swinir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/inference/inference_swinir.py -------------------------------------------------------------------------------- /options/test/BasicVSR/test_BasicVSR_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/BasicVSR/test_BasicVSR_REDS.yml -------------------------------------------------------------------------------- /options/test/BasicVSR/test_BasicVSR_Vimeo90K_BDx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/BasicVSR/test_BasicVSR_Vimeo90K_BDx4.yml -------------------------------------------------------------------------------- /options/test/BasicVSR/test_BasicVSR_Vimeo90K_BIx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/BasicVSR/test_BasicVSR_Vimeo90K_BIx4.yml -------------------------------------------------------------------------------- /options/test/BasicVSR/test_IconVSR_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/BasicVSR/test_IconVSR_REDS.yml -------------------------------------------------------------------------------- /options/test/BasicVSR/test_IconVSR_Vimeo90K_BDx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/BasicVSR/test_IconVSR_Vimeo90K_BDx4.yml -------------------------------------------------------------------------------- /options/test/BasicVSR/test_IconVSR_Vimeo90K_BIx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/BasicVSR/test_IconVSR_Vimeo90K_BIx4.yml -------------------------------------------------------------------------------- /options/test/DUF/test_DUF_official.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/DUF/test_DUF_official.yml -------------------------------------------------------------------------------- /options/test/ECBVSR/test_ECBVSR_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/ECBVSR/test_ECBVSR_REDS.yml -------------------------------------------------------------------------------- /options/test/ECBVSR/test_ECBVSR_Vimeo.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/ECBVSR/test_ECBVSR_Vimeo.yml -------------------------------------------------------------------------------- /options/test/EDSR/test_EDSR_Lx2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/EDSR/test_EDSR_Lx2.yml -------------------------------------------------------------------------------- /options/test/EDSR/test_EDSR_Lx3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/EDSR/test_EDSR_Lx3.yml -------------------------------------------------------------------------------- /options/test/EDSR/test_EDSR_Lx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/EDSR/test_EDSR_Lx4.yml -------------------------------------------------------------------------------- /options/test/EDSR/test_EDSR_Mx2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/EDSR/test_EDSR_Mx2.yml -------------------------------------------------------------------------------- /options/test/EDSR/test_EDSR_Mx3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/EDSR/test_EDSR_Mx3.yml -------------------------------------------------------------------------------- /options/test/EDSR/test_EDSR_Mx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/EDSR/test_EDSR_Mx4.yml -------------------------------------------------------------------------------- /options/test/EDVR/test_EDVR_L_deblur_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/EDVR/test_EDVR_L_deblur_REDS.yml -------------------------------------------------------------------------------- /options/test/EDVR/test_EDVR_L_deblurcomp_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/EDVR/test_EDVR_L_deblurcomp_REDS.yml -------------------------------------------------------------------------------- /options/test/EDVR/test_EDVR_L_x4_SR_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/EDVR/test_EDVR_L_x4_SR_REDS.yml -------------------------------------------------------------------------------- /options/test/EDVR/test_EDVR_L_x4_SR_Vid4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/EDVR/test_EDVR_L_x4_SR_Vid4.yml -------------------------------------------------------------------------------- /options/test/EDVR/test_EDVR_L_x4_SR_Vimeo90K.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/EDVR/test_EDVR_L_x4_SR_Vimeo90K.yml -------------------------------------------------------------------------------- /options/test/EDVR/test_EDVR_L_x4_SRblur_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/EDVR/test_EDVR_L_x4_SRblur_REDS.yml -------------------------------------------------------------------------------- /options/test/EDVR/test_EDVR_M_x4_SR_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/EDVR/test_EDVR_M_x4_SR_REDS.yml -------------------------------------------------------------------------------- /options/test/ESRGAN/test_ESRGAN_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/ESRGAN/test_ESRGAN_x4.yml -------------------------------------------------------------------------------- /options/test/ESRGAN/test_ESRGAN_x4_woGT.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/ESRGAN/test_ESRGAN_x4_woGT.yml -------------------------------------------------------------------------------- /options/test/ESRGAN/test_RRDBNet_PSNR_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/ESRGAN/test_RRDBNet_PSNR_x4.yml -------------------------------------------------------------------------------- /options/test/HiFaceGAN/test_hifacegan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/HiFaceGAN/test_hifacegan.yml -------------------------------------------------------------------------------- /options/test/HiFaceGAN/test_hifacegan_woGT.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/HiFaceGAN/test_hifacegan_woGT.yml -------------------------------------------------------------------------------- /options/test/RCAN/test_RCAN.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/RCAN/test_RCAN.yml -------------------------------------------------------------------------------- /options/test/SRResNet_SRGAN/test_MSRGAN_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/SRResNet_SRGAN/test_MSRGAN_x4.yml -------------------------------------------------------------------------------- /options/test/SRResNet_SRGAN/test_MSRResNet_x2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/SRResNet_SRGAN/test_MSRResNet_x2.yml -------------------------------------------------------------------------------- /options/test/SRResNet_SRGAN/test_MSRResNet_x3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/SRResNet_SRGAN/test_MSRResNet_x3.yml -------------------------------------------------------------------------------- /options/test/SRResNet_SRGAN/test_MSRResNet_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/SRResNet_SRGAN/test_MSRResNet_x4.yml -------------------------------------------------------------------------------- /options/test/SRResNet_SRGAN/test_MSRResNet_x4_woGT.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/SRResNet_SRGAN/test_MSRResNet_x4_woGT.yml -------------------------------------------------------------------------------- /options/test/TOF/test_TOF_official.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/test/TOF/test_TOF_official.yml -------------------------------------------------------------------------------- /options/train/BasicVSR/train_BasicVSR_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/BasicVSR/train_BasicVSR_REDS.yml -------------------------------------------------------------------------------- /options/train/BasicVSR/train_BasicVSR_Vimeo90K_BDx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/BasicVSR/train_BasicVSR_Vimeo90K_BDx4.yml -------------------------------------------------------------------------------- /options/train/BasicVSR/train_BasicVSR_Vimeo90K_BIx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/BasicVSR/train_BasicVSR_Vimeo90K_BIx4.yml -------------------------------------------------------------------------------- /options/train/BasicVSR/train_IconVSR_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/BasicVSR/train_IconVSR_REDS.yml -------------------------------------------------------------------------------- /options/train/BasicVSR/train_IconVSR_Vimeo90K_BDx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/BasicVSR/train_IconVSR_Vimeo90K_BDx4.yml -------------------------------------------------------------------------------- /options/train/BasicVSR/train_IconVSR_Vimeo90K_BIx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/BasicVSR/train_IconVSR_Vimeo90K_BIx4.yml -------------------------------------------------------------------------------- /options/train/BasicVSRPP/train_BasicVSRPP_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/BasicVSRPP/train_BasicVSRPP_REDS.yml -------------------------------------------------------------------------------- /options/train/ECBSR/train_ECBSR_x2_m4c16_prelu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/ECBSR/train_ECBSR_x2_m4c16_prelu.yml -------------------------------------------------------------------------------- /options/train/ECBSR/train_ECBSR_x4_m4c16_prelu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/ECBSR/train_ECBSR_x4_m4c16_prelu.yml -------------------------------------------------------------------------------- /options/train/ECBSR/train_ECBSR_x4_m4c16_prelu_RGB.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/ECBSR/train_ECBSR_x4_m4c16_prelu_RGB.yml -------------------------------------------------------------------------------- /options/train/ECBVSR/train_ECBVSR_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/ECBVSR/train_ECBVSR_REDS.yml -------------------------------------------------------------------------------- /options/train/ECBVSR/train_ECBVSR_Vimeo.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/ECBVSR/train_ECBVSR_Vimeo.yml -------------------------------------------------------------------------------- /options/train/EDSR/train_EDSR_Lx2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/EDSR/train_EDSR_Lx2.yml -------------------------------------------------------------------------------- /options/train/EDSR/train_EDSR_Lx3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/EDSR/train_EDSR_Lx3.yml -------------------------------------------------------------------------------- /options/train/EDSR/train_EDSR_Lx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/EDSR/train_EDSR_Lx4.yml -------------------------------------------------------------------------------- /options/train/EDSR/train_EDSR_Mx2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/EDSR/train_EDSR_Mx2.yml -------------------------------------------------------------------------------- /options/train/EDSR/train_EDSR_Mx3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/EDSR/train_EDSR_Mx3.yml -------------------------------------------------------------------------------- /options/train/EDSR/train_EDSR_Mx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/EDSR/train_EDSR_Mx4.yml -------------------------------------------------------------------------------- /options/train/EDVR/train_EDVRM_woTSA_GAN_TODO.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/EDVR/train_EDVRM_woTSA_GAN_TODO.yml -------------------------------------------------------------------------------- /options/train/EDVR/train_EDVR_L_x4_SR_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/EDVR/train_EDVR_L_x4_SR_REDS.yml -------------------------------------------------------------------------------- /options/train/EDVR/train_EDVR_L_x4_SR_REDS_woTSA.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/EDVR/train_EDVR_L_x4_SR_REDS_woTSA.yml -------------------------------------------------------------------------------- /options/train/EDVR/train_EDVR_M_x4_SR_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/EDVR/train_EDVR_M_x4_SR_REDS.yml -------------------------------------------------------------------------------- /options/train/EDVR/train_EDVR_M_x4_SR_REDS_woTSA.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/EDVR/train_EDVR_M_x4_SR_REDS_woTSA.yml -------------------------------------------------------------------------------- /options/train/ESRGAN/train_ESRGAN_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/ESRGAN/train_ESRGAN_x4.yml -------------------------------------------------------------------------------- /options/train/ESRGAN/train_RRDBNet_PSNR_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/ESRGAN/train_RRDBNet_PSNR_x4.yml -------------------------------------------------------------------------------- /options/train/HiFaceGAN/train_hifacegan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/HiFaceGAN/train_hifacegan.yml -------------------------------------------------------------------------------- /options/train/LDL/train_LDL_Real_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/LDL/train_LDL_Real_x4.yml -------------------------------------------------------------------------------- /options/train/RCAN/train_RCAN_x2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/RCAN/train_RCAN_x2.yml -------------------------------------------------------------------------------- /options/train/RealESRGAN/train_realesrgan_x2plus.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/RealESRGAN/train_realesrgan_x2plus.yml -------------------------------------------------------------------------------- /options/train/RealESRGAN/train_realesrgan_x4plus.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/RealESRGAN/train_realesrgan_x4plus.yml -------------------------------------------------------------------------------- /options/train/RealESRGAN/train_realesrnet_x2plus.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/RealESRGAN/train_realesrnet_x2plus.yml -------------------------------------------------------------------------------- /options/train/RealESRGAN/train_realesrnet_x4plus.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/RealESRGAN/train_realesrnet_x4plus.yml -------------------------------------------------------------------------------- /options/train/SRResNet_SRGAN/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/SRResNet_SRGAN/README.md -------------------------------------------------------------------------------- /options/train/SRResNet_SRGAN/train_MSRGAN_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/SRResNet_SRGAN/train_MSRGAN_x4.yml -------------------------------------------------------------------------------- /options/train/SRResNet_SRGAN/train_MSRResNet_x2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/SRResNet_SRGAN/train_MSRResNet_x2.yml -------------------------------------------------------------------------------- /options/train/SRResNet_SRGAN/train_MSRResNet_x3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/SRResNet_SRGAN/train_MSRResNet_x3.yml -------------------------------------------------------------------------------- /options/train/SRResNet_SRGAN/train_MSRResNet_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/SRResNet_SRGAN/train_MSRResNet_x4.yml -------------------------------------------------------------------------------- /options/train/StyleGAN/train_StyleGAN2_256_Cmul2_FFHQ.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/StyleGAN/train_StyleGAN2_256_Cmul2_FFHQ.yml -------------------------------------------------------------------------------- /options/train/SwinIR/train_SwinIR_SRx2_scratch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/SwinIR/train_SwinIR_SRx2_scratch.yml -------------------------------------------------------------------------------- /options/train/SwinIR/train_SwinIR_SRx4_scratch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/SwinIR/train_SwinIR_SRx4_scratch.yml -------------------------------------------------------------------------------- /options/train/VideoRecurrentGAN/train_VideoRecurrentGANModel_REDS.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/options/train/VideoRecurrentGAN/train_VideoRecurrentGANModel_REDS.yml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/data_preparation/create_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/data_preparation/create_lmdb.py -------------------------------------------------------------------------------- /scripts/data_preparation/download_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/data_preparation/download_datasets.py -------------------------------------------------------------------------------- /scripts/data_preparation/extract_images_from_tfrecords.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/data_preparation/extract_images_from_tfrecords.py -------------------------------------------------------------------------------- /scripts/data_preparation/extract_subimages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/data_preparation/extract_subimages.py -------------------------------------------------------------------------------- /scripts/data_preparation/generate_meta_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/data_preparation/generate_meta_info.py -------------------------------------------------------------------------------- /scripts/data_preparation/prepare_hifacegan_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/data_preparation/prepare_hifacegan_dataset.py -------------------------------------------------------------------------------- /scripts/data_preparation/regroup_reds_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/data_preparation/regroup_reds_dataset.py -------------------------------------------------------------------------------- /scripts/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/dist_test.sh -------------------------------------------------------------------------------- /scripts/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/dist_train.sh -------------------------------------------------------------------------------- /scripts/download_gdrive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/download_gdrive.py -------------------------------------------------------------------------------- /scripts/download_pretrained_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/download_pretrained_models.py -------------------------------------------------------------------------------- /scripts/matlab_scripts/back_projection/backprojection.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/matlab_scripts/back_projection/backprojection.m -------------------------------------------------------------------------------- /scripts/matlab_scripts/back_projection/main_bp.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/matlab_scripts/back_projection/main_bp.m -------------------------------------------------------------------------------- /scripts/matlab_scripts/back_projection/main_reverse_filter.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/matlab_scripts/back_projection/main_reverse_filter.m -------------------------------------------------------------------------------- /scripts/matlab_scripts/generate_LR_Vimeo90K.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/matlab_scripts/generate_LR_Vimeo90K.m -------------------------------------------------------------------------------- /scripts/matlab_scripts/generate_bicubic_img.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/matlab_scripts/generate_bicubic_img.m -------------------------------------------------------------------------------- /scripts/metrics/calculate_fid_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/metrics/calculate_fid_folder.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_fid_stats_from_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/metrics/calculate_fid_stats_from_datasets.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/metrics/calculate_lpips.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/metrics/calculate_niqe.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/metrics/calculate_psnr_ssim.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_stylegan2_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/metrics/calculate_stylegan2_fid.py -------------------------------------------------------------------------------- /scripts/model_conversion/convert_dfdnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/model_conversion/convert_dfdnet.py -------------------------------------------------------------------------------- /scripts/model_conversion/convert_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/model_conversion/convert_models.py -------------------------------------------------------------------------------- /scripts/model_conversion/convert_ridnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/model_conversion/convert_ridnet.py -------------------------------------------------------------------------------- /scripts/model_conversion/convert_stylegan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/model_conversion/convert_stylegan.py -------------------------------------------------------------------------------- /scripts/plot/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/plot/README.md -------------------------------------------------------------------------------- /scripts/plot/model_complexity_cmp_bsrn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/plot/model_complexity_cmp_bsrn.py -------------------------------------------------------------------------------- /scripts/publish_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/scripts/publish_models.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/setup.py -------------------------------------------------------------------------------- /test_scripts/test_discriminator_backward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/test_scripts/test_discriminator_backward.py -------------------------------------------------------------------------------- /test_scripts/test_ffhq_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/test_scripts/test_ffhq_dataset.py -------------------------------------------------------------------------------- /test_scripts/test_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/test_scripts/test_lr_scheduler.py -------------------------------------------------------------------------------- /test_scripts/test_niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/test_scripts/test_niqe.py -------------------------------------------------------------------------------- /test_scripts/test_paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/test_scripts/test_paired_image_dataset.py -------------------------------------------------------------------------------- /test_scripts/test_reds_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/test_scripts/test_reds_dataset.py -------------------------------------------------------------------------------- /test_scripts/test_vimeo90k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/test_scripts/test_vimeo90k_dataset.py -------------------------------------------------------------------------------- /tests/README.md: -------------------------------------------------------------------------------- 1 | # UnitTest 2 | 3 | - It requires GPU CUDA environment 4 | -------------------------------------------------------------------------------- /tests/data/gt.lmdb/data.mdb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/data/gt.lmdb/data.mdb -------------------------------------------------------------------------------- /tests/data/gt.lmdb/lock.mdb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/data/gt.lmdb/lock.mdb -------------------------------------------------------------------------------- /tests/data/gt.lmdb/meta_info.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/data/gt.lmdb/meta_info.txt -------------------------------------------------------------------------------- /tests/data/lq.lmdb/data.mdb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/data/lq.lmdb/data.mdb -------------------------------------------------------------------------------- /tests/data/lq.lmdb/lock.mdb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/data/lq.lmdb/lock.mdb -------------------------------------------------------------------------------- /tests/data/lq.lmdb/meta_info.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/data/lq.lmdb/meta_info.txt -------------------------------------------------------------------------------- /tests/data/meta_info_gt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/data/meta_info_gt.txt -------------------------------------------------------------------------------- /tests/data/meta_info_pair.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/data/meta_info_pair.txt -------------------------------------------------------------------------------- /tests/test_archs/test_basicvsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/test_archs/test_basicvsr_arch.py -------------------------------------------------------------------------------- /tests/test_archs/test_discriminator_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/test_archs/test_discriminator_arch.py -------------------------------------------------------------------------------- /tests/test_archs/test_duf_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/test_archs/test_duf_arch.py -------------------------------------------------------------------------------- /tests/test_archs/test_ecbsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/test_archs/test_ecbsr_arch.py -------------------------------------------------------------------------------- /tests/test_archs/test_srresnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/test_archs/test_srresnet_arch.py -------------------------------------------------------------------------------- /tests/test_data/test_paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/test_data/test_paired_image_dataset.py -------------------------------------------------------------------------------- /tests/test_data/test_single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/test_data/test_single_image_dataset.py -------------------------------------------------------------------------------- /tests/test_losses/test_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/test_losses/test_losses.py -------------------------------------------------------------------------------- /tests/test_metrics/test_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/test_metrics/test_psnr_ssim.py -------------------------------------------------------------------------------- /tests/test_models/test_sr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kimsure/KSNet/HEAD/tests/test_models/test_sr_model.py --------------------------------------------------------------------------------