├── .gitignore ├── LICENSE ├── README.md ├── VERSION ├── basicsr ├── __init__.py ├── archs │ ├── __init__.py │ ├── arch_util.py │ ├── hit_sir_arch.py │ ├── hit_sng_arch.py │ ├── hit_srf_arch.py │ └── vgg_arch.py ├── data │ ├── __init__.py │ ├── data_sampler.py │ ├── data_util.py │ ├── paired_image_dataset.py │ ├── prefetch_dataloader.py │ ├── single_image_dataset.py │ └── transforms.py ├── losses │ ├── __init__.py │ ├── loss_util.py │ └── losses.py ├── metrics │ ├── __init__.py │ ├── metric_util.py │ └── psnr_ssim.py ├── models │ ├── __init__.py │ ├── base_model.py │ ├── hit_model.py │ ├── lr_scheduler.py │ └── sr_model.py ├── test.py ├── train.py └── utils │ ├── __init__.py │ ├── dist_util.py │ ├── file_client.py │ ├── img_util.py │ ├── logger.py │ ├── matlab_functions.py │ ├── misc.py │ ├── options.py │ └── registry.py ├── datasets ├── README.md └── single │ ├── test_1.png │ └── test_2.png ├── experiments ├── README.md └── pretrained_models │ └── README.md ├── figs ├── HiT-SR.png ├── LAM.png ├── Quali-main.png ├── Quali-supp1.png ├── Quali-supp2.png ├── convergence-comparison.png ├── efficiency-comparison.png ├── framework.png ├── logo.png ├── overall_improvements.png ├── performance-comparison.png └── quantitative-comparison.png ├── options ├── Test │ ├── test_HiT_SIR_x2.yml │ ├── test_HiT_SIR_x3.yml │ ├── test_HiT_SIR_x4.yml │ ├── test_HiT_SNG_x2.yml │ ├── test_HiT_SNG_x3.yml │ ├── test_HiT_SNG_x4.yml │ ├── test_HiT_SRF_x2.yml │ ├── test_HiT_SRF_x2_DF2K.yml │ ├── test_HiT_SRF_x3.yml │ ├── test_HiT_SRF_x3_DF2K.yml │ ├── test_HiT_SRF_x4.yml │ ├── test_HiT_SRF_x4_DF2K.yml │ ├── test_single_x2.yml │ ├── test_single_x3.yml │ └── test_single_x4.yml └── Train │ ├── train_HiT_SIR_x2.yml │ ├── train_HiT_SIR_x3.yml │ ├── train_HiT_SIR_x4.yml │ ├── train_HiT_SNG_x2.yml │ ├── train_HiT_SNG_x3.yml │ ├── train_HiT_SNG_x4.yml │ ├── train_HiT_SRF_x2.yml │ ├── train_HiT_SRF_x2_DF2K.yml │ ├── train_HiT_SRF_x3.yml │ ├── train_HiT_SRF_x3_DF2K.yml │ ├── train_HiT_SRF_x4.yml │ └── train_HiT_SRF_x4_DF2K.yml ├── requirements.txt ├── results └── README.md └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/README.md -------------------------------------------------------------------------------- /VERSION: -------------------------------------------------------------------------------- 1 | 1.3.5 2 | -------------------------------------------------------------------------------- /basicsr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/archs/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/archs/arch_util.py -------------------------------------------------------------------------------- /basicsr/archs/hit_sir_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/archs/hit_sir_arch.py -------------------------------------------------------------------------------- /basicsr/archs/hit_sng_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/archs/hit_sng_arch.py -------------------------------------------------------------------------------- /basicsr/archs/hit_srf_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/archs/hit_srf_arch.py -------------------------------------------------------------------------------- /basicsr/archs/vgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/archs/vgg_arch.py -------------------------------------------------------------------------------- /basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/data/__init__.py -------------------------------------------------------------------------------- /basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/data/data_util.py -------------------------------------------------------------------------------- /basicsr/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/data/paired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /basicsr/data/single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/data/single_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/data/transforms.py -------------------------------------------------------------------------------- /basicsr/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/losses/__init__.py -------------------------------------------------------------------------------- /basicsr/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/losses/loss_util.py -------------------------------------------------------------------------------- /basicsr/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/losses/losses.py -------------------------------------------------------------------------------- /basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/models/__init__.py -------------------------------------------------------------------------------- /basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/models/base_model.py -------------------------------------------------------------------------------- /basicsr/models/hit_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/models/hit_model.py -------------------------------------------------------------------------------- /basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /basicsr/models/sr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/models/sr_model.py -------------------------------------------------------------------------------- /basicsr/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/test.py -------------------------------------------------------------------------------- /basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/train.py -------------------------------------------------------------------------------- /basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/utils/file_client.py -------------------------------------------------------------------------------- /basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/utils/img_util.py -------------------------------------------------------------------------------- /basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/utils/logger.py -------------------------------------------------------------------------------- /basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/utils/misc.py -------------------------------------------------------------------------------- /basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/utils/options.py -------------------------------------------------------------------------------- /basicsr/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/basicsr/utils/registry.py -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/datasets/README.md -------------------------------------------------------------------------------- /datasets/single/test_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/datasets/single/test_1.png -------------------------------------------------------------------------------- /datasets/single/test_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/datasets/single/test_2.png -------------------------------------------------------------------------------- /experiments/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/experiments/README.md -------------------------------------------------------------------------------- /experiments/pretrained_models/README.md: -------------------------------------------------------------------------------- 1 | Place pretrained models here. -------------------------------------------------------------------------------- /figs/HiT-SR.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/figs/HiT-SR.png -------------------------------------------------------------------------------- /figs/LAM.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/figs/LAM.png -------------------------------------------------------------------------------- /figs/Quali-main.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/figs/Quali-main.png -------------------------------------------------------------------------------- /figs/Quali-supp1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/figs/Quali-supp1.png -------------------------------------------------------------------------------- /figs/Quali-supp2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/figs/Quali-supp2.png -------------------------------------------------------------------------------- /figs/convergence-comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/figs/convergence-comparison.png -------------------------------------------------------------------------------- /figs/efficiency-comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/figs/efficiency-comparison.png -------------------------------------------------------------------------------- /figs/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/figs/framework.png -------------------------------------------------------------------------------- /figs/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/figs/logo.png -------------------------------------------------------------------------------- /figs/overall_improvements.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/figs/overall_improvements.png -------------------------------------------------------------------------------- /figs/performance-comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/figs/performance-comparison.png -------------------------------------------------------------------------------- /figs/quantitative-comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/figs/quantitative-comparison.png -------------------------------------------------------------------------------- /options/Test/test_HiT_SIR_x2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_HiT_SIR_x2.yml -------------------------------------------------------------------------------- /options/Test/test_HiT_SIR_x3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_HiT_SIR_x3.yml -------------------------------------------------------------------------------- /options/Test/test_HiT_SIR_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_HiT_SIR_x4.yml -------------------------------------------------------------------------------- /options/Test/test_HiT_SNG_x2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_HiT_SNG_x2.yml -------------------------------------------------------------------------------- /options/Test/test_HiT_SNG_x3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_HiT_SNG_x3.yml -------------------------------------------------------------------------------- /options/Test/test_HiT_SNG_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_HiT_SNG_x4.yml -------------------------------------------------------------------------------- /options/Test/test_HiT_SRF_x2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_HiT_SRF_x2.yml -------------------------------------------------------------------------------- /options/Test/test_HiT_SRF_x2_DF2K.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_HiT_SRF_x2_DF2K.yml -------------------------------------------------------------------------------- /options/Test/test_HiT_SRF_x3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_HiT_SRF_x3.yml -------------------------------------------------------------------------------- /options/Test/test_HiT_SRF_x3_DF2K.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_HiT_SRF_x3_DF2K.yml -------------------------------------------------------------------------------- /options/Test/test_HiT_SRF_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_HiT_SRF_x4.yml -------------------------------------------------------------------------------- /options/Test/test_HiT_SRF_x4_DF2K.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_HiT_SRF_x4_DF2K.yml -------------------------------------------------------------------------------- /options/Test/test_single_x2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_single_x2.yml -------------------------------------------------------------------------------- /options/Test/test_single_x3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_single_x3.yml -------------------------------------------------------------------------------- /options/Test/test_single_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Test/test_single_x4.yml -------------------------------------------------------------------------------- /options/Train/train_HiT_SIR_x2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Train/train_HiT_SIR_x2.yml -------------------------------------------------------------------------------- /options/Train/train_HiT_SIR_x3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Train/train_HiT_SIR_x3.yml -------------------------------------------------------------------------------- /options/Train/train_HiT_SIR_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Train/train_HiT_SIR_x4.yml -------------------------------------------------------------------------------- /options/Train/train_HiT_SNG_x2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Train/train_HiT_SNG_x2.yml -------------------------------------------------------------------------------- /options/Train/train_HiT_SNG_x3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Train/train_HiT_SNG_x3.yml -------------------------------------------------------------------------------- /options/Train/train_HiT_SNG_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Train/train_HiT_SNG_x4.yml -------------------------------------------------------------------------------- /options/Train/train_HiT_SRF_x2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Train/train_HiT_SRF_x2.yml -------------------------------------------------------------------------------- /options/Train/train_HiT_SRF_x2_DF2K.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Train/train_HiT_SRF_x2_DF2K.yml -------------------------------------------------------------------------------- /options/Train/train_HiT_SRF_x3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Train/train_HiT_SRF_x3.yml -------------------------------------------------------------------------------- /options/Train/train_HiT_SRF_x3_DF2K.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Train/train_HiT_SRF_x3_DF2K.yml -------------------------------------------------------------------------------- /options/Train/train_HiT_SRF_x4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Train/train_HiT_SRF_x4.yml -------------------------------------------------------------------------------- /options/Train/train_HiT_SRF_x4_DF2K.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/options/Train/train_HiT_SRF_x4_DF2K.yml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/README.md: -------------------------------------------------------------------------------- 1 | The testing results. 2 | 3 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangZ-0/HiT-SR/HEAD/setup.py --------------------------------------------------------------------------------