├── .gitignore ├── LICENSE.txt ├── README.md ├── VERSION ├── basicsr ├── __init__.py ├── archs │ ├── __init__.py │ ├── arch_util.py │ ├── srformer_arch.py │ ├── srformerv2_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 ├── infer_sr.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 ├── figs ├── classicalSR_1.png ├── classicalSR_2.png ├── lightweightSR_1.png ├── lightweightSR_2.png ├── model_size.png ├── psa.png ├── realworld.png └── simple_compare.png ├── options ├── test │ ├── SRFormer │ │ ├── test_SRFormer-S_x4_real.yml │ │ ├── test_SRFormer_DF2Ksrx2.yml │ │ ├── test_SRFormer_DF2Ksrx3.yml │ │ ├── test_SRFormer_DF2Ksrx4.yml │ │ ├── test_SRFormer_light_DIV2Ksrx2.yml │ │ ├── test_SRFormer_light_DIV2Ksrx3.yml │ │ └── test_SRFormer_light_DIV2Ksrx4.yml │ └── SRFormerV2 │ │ └── test_SRFormer_DF2Ksrx4.yml └── train │ └── SRFormer │ ├── train_SRFormer_SRx2_scratch.yml │ ├── train_SRFormer_SRx3_scratch.yml │ ├── train_SRFormer_SRx4_scratch.yml │ ├── train_SRFormer_light_SRx2_scratch.yml │ ├── train_SRFormer_light_SRx3_scratch.yml │ └── train_SRFormer_light_SRx4_scratch.yml ├── qad ├── 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 ├── data │ └── baboon.png ├── 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/HVision-NKU/SRFormer/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/README.md -------------------------------------------------------------------------------- /VERSION: -------------------------------------------------------------------------------- 1 | 1.4.2 2 | -------------------------------------------------------------------------------- /basicsr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/archs/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/archs/arch_util.py -------------------------------------------------------------------------------- /basicsr/archs/srformer_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/archs/srformer_arch.py -------------------------------------------------------------------------------- /basicsr/archs/srformerv2_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/archs/srformerv2_arch.py -------------------------------------------------------------------------------- /basicsr/archs/vgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/archs/vgg_arch.py -------------------------------------------------------------------------------- /basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/__init__.py -------------------------------------------------------------------------------- /basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/data_util.py -------------------------------------------------------------------------------- /basicsr/data/degradations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/degradations.py -------------------------------------------------------------------------------- /basicsr/data/ffhq_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/ffhq_dataset.py -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_DIV2K800sub_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/meta_info/meta_info_DIV2K800sub_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDS4_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/meta_info/meta_info_REDS4_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDS_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/meta_info/meta_info_REDS_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDSofficial4_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/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/HVision-NKU/SRFormer/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/HVision-NKU/SRFormer/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/HVision-NKU/SRFormer/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/HVision-NKU/SRFormer/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/HVision-NKU/SRFormer/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/HVision-NKU/SRFormer/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_train_GT.txt -------------------------------------------------------------------------------- /basicsr/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/paired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /basicsr/data/realesrgan_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/realesrgan_dataset.py -------------------------------------------------------------------------------- /basicsr/data/realesrgan_paired_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/realesrgan_paired_dataset.py -------------------------------------------------------------------------------- /basicsr/data/reds_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/reds_dataset.py -------------------------------------------------------------------------------- /basicsr/data/single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/single_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/transforms.py -------------------------------------------------------------------------------- /basicsr/data/video_test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/video_test_dataset.py -------------------------------------------------------------------------------- /basicsr/data/vimeo90k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/data/vimeo90k_dataset.py -------------------------------------------------------------------------------- /basicsr/infer_sr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/infer_sr.py -------------------------------------------------------------------------------- /basicsr/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/losses/__init__.py -------------------------------------------------------------------------------- /basicsr/losses/basic_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/losses/basic_loss.py -------------------------------------------------------------------------------- /basicsr/losses/gan_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/losses/gan_loss.py -------------------------------------------------------------------------------- /basicsr/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/losses/loss_util.py -------------------------------------------------------------------------------- /basicsr/metrics/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/metrics/README.md -------------------------------------------------------------------------------- /basicsr/metrics/README_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/metrics/README_CN.md -------------------------------------------------------------------------------- /basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /basicsr/metrics/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/metrics/fid.py -------------------------------------------------------------------------------- /basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/metrics/niqe.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe_pris_params.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/metrics/niqe_pris_params.npz -------------------------------------------------------------------------------- /basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/metrics/test_metrics/test_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/metrics/test_metrics/test_psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/__init__.py -------------------------------------------------------------------------------- /basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/base_model.py -------------------------------------------------------------------------------- /basicsr/models/edvr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/edvr_model.py -------------------------------------------------------------------------------- /basicsr/models/esrgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/esrgan_model.py -------------------------------------------------------------------------------- /basicsr/models/hifacegan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/hifacegan_model.py -------------------------------------------------------------------------------- /basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /basicsr/models/realesrgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/realesrgan_model.py -------------------------------------------------------------------------------- /basicsr/models/realesrnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/realesrnet_model.py -------------------------------------------------------------------------------- /basicsr/models/sr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/sr_model.py -------------------------------------------------------------------------------- /basicsr/models/srgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/srgan_model.py -------------------------------------------------------------------------------- /basicsr/models/stylegan2_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/stylegan2_model.py -------------------------------------------------------------------------------- /basicsr/models/swinir_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/swinir_model.py -------------------------------------------------------------------------------- /basicsr/models/video_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/video_base_model.py -------------------------------------------------------------------------------- /basicsr/models/video_gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/video_gan_model.py -------------------------------------------------------------------------------- /basicsr/models/video_recurrent_gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/video_recurrent_gan_model.py -------------------------------------------------------------------------------- /basicsr/models/video_recurrent_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/models/video_recurrent_model.py -------------------------------------------------------------------------------- /basicsr/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /basicsr/ops/dcn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/ops/dcn/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/dcn/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/ops/dcn/deform_conv.py -------------------------------------------------------------------------------- /basicsr/ops/dcn/src/deform_conv_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/ops/dcn/src/deform_conv_cuda.cpp -------------------------------------------------------------------------------- /basicsr/ops/dcn/src/deform_conv_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/ops/dcn/src/deform_conv_cuda_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/dcn/src/deform_conv_ext.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/ops/dcn/src/deform_conv_ext.cpp -------------------------------------------------------------------------------- /basicsr/ops/fused_act/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/ops/fused_act/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/fused_act/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/ops/fused_act/fused_act.py -------------------------------------------------------------------------------- /basicsr/ops/fused_act/src/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/ops/fused_act/src/fused_bias_act.cpp -------------------------------------------------------------------------------- /basicsr/ops/fused_act/src/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/ops/fused_act/src/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/ops/upfirdn2d/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/src/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/ops/upfirdn2d/src/upfirdn2d.cpp -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/src/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/ops/upfirdn2d/src/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/ops/upfirdn2d/upfirdn2d.py -------------------------------------------------------------------------------- /basicsr/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/test.py -------------------------------------------------------------------------------- /basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/train.py -------------------------------------------------------------------------------- /basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/color_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/color_util.py -------------------------------------------------------------------------------- /basicsr/utils/diffjpeg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/diffjpeg.py -------------------------------------------------------------------------------- /basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /basicsr/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/download_util.py -------------------------------------------------------------------------------- /basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/file_client.py -------------------------------------------------------------------------------- /basicsr/utils/flow_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/flow_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_process_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/img_process_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/img_util.py -------------------------------------------------------------------------------- /basicsr/utils/lmdb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/lmdb_util.py -------------------------------------------------------------------------------- /basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/logger.py -------------------------------------------------------------------------------- /basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/misc.py -------------------------------------------------------------------------------- /basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/options.py -------------------------------------------------------------------------------- /basicsr/utils/plot_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/plot_util.py -------------------------------------------------------------------------------- /basicsr/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/basicsr/utils/registry.py -------------------------------------------------------------------------------- /figs/classicalSR_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/figs/classicalSR_1.png -------------------------------------------------------------------------------- /figs/classicalSR_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/figs/classicalSR_2.png -------------------------------------------------------------------------------- /figs/lightweightSR_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/figs/lightweightSR_1.png -------------------------------------------------------------------------------- /figs/lightweightSR_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/figs/lightweightSR_2.png -------------------------------------------------------------------------------- /figs/model_size.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/figs/model_size.png -------------------------------------------------------------------------------- /figs/psa.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/figs/psa.png -------------------------------------------------------------------------------- /figs/realworld.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/figs/realworld.png -------------------------------------------------------------------------------- /figs/simple_compare.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/figs/simple_compare.png -------------------------------------------------------------------------------- /options/test/SRFormer/test_SRFormer-S_x4_real.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/options/test/SRFormer/test_SRFormer-S_x4_real.yml -------------------------------------------------------------------------------- /options/test/SRFormer/test_SRFormer_DF2Ksrx2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/options/test/SRFormer/test_SRFormer_DF2Ksrx2.yml -------------------------------------------------------------------------------- /options/test/SRFormer/test_SRFormer_DF2Ksrx3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/options/test/SRFormer/test_SRFormer_DF2Ksrx3.yml -------------------------------------------------------------------------------- /options/test/SRFormer/test_SRFormer_DF2Ksrx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/options/test/SRFormer/test_SRFormer_DF2Ksrx4.yml -------------------------------------------------------------------------------- /options/test/SRFormer/test_SRFormer_light_DIV2Ksrx2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/options/test/SRFormer/test_SRFormer_light_DIV2Ksrx2.yml -------------------------------------------------------------------------------- /options/test/SRFormer/test_SRFormer_light_DIV2Ksrx3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/options/test/SRFormer/test_SRFormer_light_DIV2Ksrx3.yml -------------------------------------------------------------------------------- /options/test/SRFormer/test_SRFormer_light_DIV2Ksrx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/options/test/SRFormer/test_SRFormer_light_DIV2Ksrx4.yml -------------------------------------------------------------------------------- /options/test/SRFormerV2/test_SRFormer_DF2Ksrx4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/options/test/SRFormerV2/test_SRFormer_DF2Ksrx4.yml -------------------------------------------------------------------------------- /options/train/SRFormer/train_SRFormer_SRx2_scratch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/options/train/SRFormer/train_SRFormer_SRx2_scratch.yml -------------------------------------------------------------------------------- /options/train/SRFormer/train_SRFormer_SRx3_scratch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/options/train/SRFormer/train_SRFormer_SRx3_scratch.yml -------------------------------------------------------------------------------- /options/train/SRFormer/train_SRFormer_SRx4_scratch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/options/train/SRFormer/train_SRFormer_SRx4_scratch.yml -------------------------------------------------------------------------------- /options/train/SRFormer/train_SRFormer_light_SRx2_scratch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/options/train/SRFormer/train_SRFormer_light_SRx2_scratch.yml -------------------------------------------------------------------------------- /options/train/SRFormer/train_SRFormer_light_SRx3_scratch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/options/train/SRFormer/train_SRFormer_light_SRx3_scratch.yml -------------------------------------------------------------------------------- /options/train/SRFormer/train_SRFormer_light_SRx4_scratch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/options/train/SRFormer/train_SRFormer_light_SRx4_scratch.yml -------------------------------------------------------------------------------- /qad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/qad -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/data_preparation/create_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/data_preparation/create_lmdb.py -------------------------------------------------------------------------------- /scripts/data_preparation/download_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/data_preparation/download_datasets.py -------------------------------------------------------------------------------- /scripts/data_preparation/extract_images_from_tfrecords.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/data_preparation/extract_images_from_tfrecords.py -------------------------------------------------------------------------------- /scripts/data_preparation/extract_subimages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/data_preparation/extract_subimages.py -------------------------------------------------------------------------------- /scripts/data_preparation/generate_meta_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/data_preparation/generate_meta_info.py -------------------------------------------------------------------------------- /scripts/data_preparation/prepare_hifacegan_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/data_preparation/prepare_hifacegan_dataset.py -------------------------------------------------------------------------------- /scripts/data_preparation/regroup_reds_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/data_preparation/regroup_reds_dataset.py -------------------------------------------------------------------------------- /scripts/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/dist_test.sh -------------------------------------------------------------------------------- /scripts/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/dist_train.sh -------------------------------------------------------------------------------- /scripts/download_gdrive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/download_gdrive.py -------------------------------------------------------------------------------- /scripts/download_pretrained_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/download_pretrained_models.py -------------------------------------------------------------------------------- /scripts/matlab_scripts/back_projection/backprojection.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/matlab_scripts/back_projection/backprojection.m -------------------------------------------------------------------------------- /scripts/matlab_scripts/back_projection/main_bp.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/matlab_scripts/back_projection/main_bp.m -------------------------------------------------------------------------------- /scripts/matlab_scripts/back_projection/main_reverse_filter.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/matlab_scripts/back_projection/main_reverse_filter.m -------------------------------------------------------------------------------- /scripts/matlab_scripts/generate_LR_Vimeo90K.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/matlab_scripts/generate_LR_Vimeo90K.m -------------------------------------------------------------------------------- /scripts/matlab_scripts/generate_bicubic_img.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/matlab_scripts/generate_bicubic_img.m -------------------------------------------------------------------------------- /scripts/metrics/calculate_fid_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/metrics/calculate_fid_folder.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_fid_stats_from_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/metrics/calculate_fid_stats_from_datasets.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/metrics/calculate_lpips.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/metrics/calculate_niqe.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/metrics/calculate_psnr_ssim.py -------------------------------------------------------------------------------- /scripts/metrics/calculate_stylegan2_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/metrics/calculate_stylegan2_fid.py -------------------------------------------------------------------------------- /scripts/model_conversion/convert_dfdnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/model_conversion/convert_dfdnet.py -------------------------------------------------------------------------------- /scripts/model_conversion/convert_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/model_conversion/convert_models.py -------------------------------------------------------------------------------- /scripts/model_conversion/convert_ridnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/model_conversion/convert_ridnet.py -------------------------------------------------------------------------------- /scripts/model_conversion/convert_stylegan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/model_conversion/convert_stylegan.py -------------------------------------------------------------------------------- /scripts/plot/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/plot/README.md -------------------------------------------------------------------------------- /scripts/plot/model_complexity_cmp_bsrn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/plot/model_complexity_cmp_bsrn.py -------------------------------------------------------------------------------- /scripts/publish_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/scripts/publish_models.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/setup.py -------------------------------------------------------------------------------- /test_scripts/data/baboon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/test_scripts/data/baboon.png -------------------------------------------------------------------------------- /test_scripts/test_discriminator_backward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/test_scripts/test_discriminator_backward.py -------------------------------------------------------------------------------- /test_scripts/test_ffhq_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/test_scripts/test_ffhq_dataset.py -------------------------------------------------------------------------------- /test_scripts/test_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/test_scripts/test_lr_scheduler.py -------------------------------------------------------------------------------- /test_scripts/test_niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/test_scripts/test_niqe.py -------------------------------------------------------------------------------- /test_scripts/test_paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/test_scripts/test_paired_image_dataset.py -------------------------------------------------------------------------------- /test_scripts/test_reds_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/test_scripts/test_reds_dataset.py -------------------------------------------------------------------------------- /test_scripts/test_vimeo90k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/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/HVision-NKU/SRFormer/HEAD/tests/data/gt.lmdb/data.mdb -------------------------------------------------------------------------------- /tests/data/gt.lmdb/lock.mdb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/data/gt.lmdb/lock.mdb -------------------------------------------------------------------------------- /tests/data/gt.lmdb/meta_info.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/data/gt.lmdb/meta_info.txt -------------------------------------------------------------------------------- /tests/data/lq.lmdb/data.mdb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/data/lq.lmdb/data.mdb -------------------------------------------------------------------------------- /tests/data/lq.lmdb/lock.mdb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/data/lq.lmdb/lock.mdb -------------------------------------------------------------------------------- /tests/data/lq.lmdb/meta_info.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/data/lq.lmdb/meta_info.txt -------------------------------------------------------------------------------- /tests/data/meta_info_gt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/data/meta_info_gt.txt -------------------------------------------------------------------------------- /tests/data/meta_info_pair.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/data/meta_info_pair.txt -------------------------------------------------------------------------------- /tests/test_archs/test_basicvsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/test_archs/test_basicvsr_arch.py -------------------------------------------------------------------------------- /tests/test_archs/test_discriminator_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/test_archs/test_discriminator_arch.py -------------------------------------------------------------------------------- /tests/test_archs/test_duf_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/test_archs/test_duf_arch.py -------------------------------------------------------------------------------- /tests/test_archs/test_ecbsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/test_archs/test_ecbsr_arch.py -------------------------------------------------------------------------------- /tests/test_archs/test_srresnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/test_archs/test_srresnet_arch.py -------------------------------------------------------------------------------- /tests/test_data/test_paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/test_data/test_paired_image_dataset.py -------------------------------------------------------------------------------- /tests/test_data/test_single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/test_data/test_single_image_dataset.py -------------------------------------------------------------------------------- /tests/test_losses/test_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/test_losses/test_losses.py -------------------------------------------------------------------------------- /tests/test_metrics/test_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/test_metrics/test_psnr_ssim.py -------------------------------------------------------------------------------- /tests/test_models/test_sr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/SRFormer/HEAD/tests/test_models/test_sr_model.py --------------------------------------------------------------------------------