├── .gitignore ├── LICENSE ├── README.md ├── data ├── __init__.py ├── dataset_SR_guided.py └── select_dataset.py ├── images ├── archi.png ├── example1.png ├── example2.png ├── example3.png └── slp.png ├── main_train_SwinFuSR.py ├── models ├── basicblock.py ├── discriminator.py ├── loss.py ├── loss_sr.py ├── loss_ssim.py ├── loss_vif.py ├── model_base.py ├── model_plain.py ├── network_feature.py ├── network_swinfusionSR.py ├── op │ ├── __init__.py │ ├── fused_act.py │ ├── fused_bias_act.cpp │ ├── fused_bias_act_kernel.cu │ ├── upfirdn2d.cpp │ ├── upfirdn2d.py │ └── upfirdn2d_kernel.cu ├── select_model.py └── select_network.py ├── options ├── test_swinFuSR.json ├── train_augmentation.json ├── train_augmentation_finetune.json ├── train_baseline.json ├── train_baseline_finetune.json ├── train_final.json ├── train_final_fine.json └── train_x16.json ├── requirements.txt ├── test_SwinFuSR.py └── utils ├── utils_bnorm.py ├── utils_color.py ├── utils_dist.py ├── utils_googledownload.py ├── utils_image.py ├── utils_logger.py ├── utils_mat.py ├── utils_matconvnet.py ├── utils_model.py ├── utils_modelsummary.py ├── utils_option.py ├── utils_params.py ├── utils_receptivefield.py ├── utils_regularizers.py └── utils_sisr.py /.gitignore: -------------------------------------------------------------------------------- 1 | /Model/ 2 | *pycache* 3 | /Test/ 4 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /data/dataset_SR_guided.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/data/dataset_SR_guided.py -------------------------------------------------------------------------------- /data/select_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/data/select_dataset.py -------------------------------------------------------------------------------- /images/archi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/images/archi.png -------------------------------------------------------------------------------- /images/example1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/images/example1.png -------------------------------------------------------------------------------- /images/example2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/images/example2.png -------------------------------------------------------------------------------- /images/example3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/images/example3.png -------------------------------------------------------------------------------- /images/slp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/images/slp.png -------------------------------------------------------------------------------- /main_train_SwinFuSR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/main_train_SwinFuSR.py -------------------------------------------------------------------------------- /models/basicblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/basicblock.py -------------------------------------------------------------------------------- /models/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/discriminator.py -------------------------------------------------------------------------------- /models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/loss.py -------------------------------------------------------------------------------- /models/loss_sr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/loss_sr.py -------------------------------------------------------------------------------- /models/loss_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/loss_ssim.py -------------------------------------------------------------------------------- /models/loss_vif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/loss_vif.py -------------------------------------------------------------------------------- /models/model_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/model_base.py -------------------------------------------------------------------------------- /models/model_plain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/model_plain.py -------------------------------------------------------------------------------- /models/network_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/network_feature.py -------------------------------------------------------------------------------- /models/network_swinfusionSR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/network_swinfusionSR.py -------------------------------------------------------------------------------- /models/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/op/__init__.py -------------------------------------------------------------------------------- /models/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/op/fused_act.py -------------------------------------------------------------------------------- /models/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /models/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /models/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /models/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/op/upfirdn2d.py -------------------------------------------------------------------------------- /models/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /models/select_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/select_model.py -------------------------------------------------------------------------------- /models/select_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/models/select_network.py -------------------------------------------------------------------------------- /options/test_swinFuSR.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/options/test_swinFuSR.json -------------------------------------------------------------------------------- /options/train_augmentation.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/options/train_augmentation.json -------------------------------------------------------------------------------- /options/train_augmentation_finetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/options/train_augmentation_finetune.json -------------------------------------------------------------------------------- /options/train_baseline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/options/train_baseline.json -------------------------------------------------------------------------------- /options/train_baseline_finetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/options/train_baseline_finetune.json -------------------------------------------------------------------------------- /options/train_final.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/options/train_final.json -------------------------------------------------------------------------------- /options/train_final_fine.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/options/train_final_fine.json -------------------------------------------------------------------------------- /options/train_x16.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/options/train_x16.json -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/requirements.txt -------------------------------------------------------------------------------- /test_SwinFuSR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/test_SwinFuSR.py -------------------------------------------------------------------------------- /utils/utils_bnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_bnorm.py -------------------------------------------------------------------------------- /utils/utils_color.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_color.py -------------------------------------------------------------------------------- /utils/utils_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_dist.py -------------------------------------------------------------------------------- /utils/utils_googledownload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_googledownload.py -------------------------------------------------------------------------------- /utils/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_image.py -------------------------------------------------------------------------------- /utils/utils_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_logger.py -------------------------------------------------------------------------------- /utils/utils_mat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_mat.py -------------------------------------------------------------------------------- /utils/utils_matconvnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_matconvnet.py -------------------------------------------------------------------------------- /utils/utils_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_model.py -------------------------------------------------------------------------------- /utils/utils_modelsummary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_modelsummary.py -------------------------------------------------------------------------------- /utils/utils_option.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_option.py -------------------------------------------------------------------------------- /utils/utils_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_params.py -------------------------------------------------------------------------------- /utils/utils_receptivefield.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_receptivefield.py -------------------------------------------------------------------------------- /utils/utils_regularizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_regularizers.py -------------------------------------------------------------------------------- /utils/utils_sisr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VisionICLab/SwinFuSR/HEAD/utils/utils_sisr.py --------------------------------------------------------------------------------