├── .idea ├── CSD.iml ├── inspectionProfiles │ ├── Project_Default.xml │ └── profiles_settings.xml ├── modules.xml └── workspace.xml ├── MindSpore version ├── README.md ├── csd_train.py ├── eval.py ├── export.py ├── src │ ├── args.py │ ├── common.py │ ├── config.py │ ├── contras_loss.py │ ├── data │ │ ├── common.py │ │ ├── div2k.py │ │ └── srdata.py │ ├── edsr_model.py │ ├── edsr_slim.py │ ├── metric.py │ ├── metrics.py │ ├── rcan_model.py │ └── vgg_model.py ├── train.py └── utils │ ├── __init__.py │ └── var_init.py ├── PyTorch version ├── README.md ├── data │ ├── __init__.py │ ├── benchmark.py │ ├── bsd500.py │ ├── common.py │ ├── demo.py │ ├── div2k.py │ ├── div2kjpeg.py │ ├── sr291.py │ ├── srdata.py │ └── video.py ├── dataloader.py ├── loss │ ├── __init__.py │ ├── adversarial.py │ ├── contrast_loss.py │ ├── discriminator.py │ ├── perceptual.py │ └── vgg.py ├── main.py ├── model │ ├── __init__.py │ ├── carn.py │ ├── common.py │ ├── edsr.py │ └── rcan.py ├── option.py ├── trainer │ └── slim_contrast_trainer.py └── utils │ ├── __init__.py │ ├── niqe.py │ ├── niqe_image_params.mat │ ├── spatial_trans.py │ ├── ssim.py │ └── utility.py ├── README.md └── images ├── debug.log ├── model.png ├── psnr-speed.png ├── psnr-tradeoff.png ├── table.png ├── tradeoff.png └── visual.png /.idea/CSD.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/.idea/CSD.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/Project_Default.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/.idea/inspectionProfiles/Project_Default.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/workspace.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/.idea/workspace.xml -------------------------------------------------------------------------------- /MindSpore version/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/README.md -------------------------------------------------------------------------------- /MindSpore version/csd_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/csd_train.py -------------------------------------------------------------------------------- /MindSpore version/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/eval.py -------------------------------------------------------------------------------- /MindSpore version/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/export.py -------------------------------------------------------------------------------- /MindSpore version/src/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/src/args.py -------------------------------------------------------------------------------- /MindSpore version/src/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/src/common.py -------------------------------------------------------------------------------- /MindSpore version/src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/src/config.py -------------------------------------------------------------------------------- /MindSpore version/src/contras_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/src/contras_loss.py -------------------------------------------------------------------------------- /MindSpore version/src/data/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/src/data/common.py -------------------------------------------------------------------------------- /MindSpore version/src/data/div2k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/src/data/div2k.py -------------------------------------------------------------------------------- /MindSpore version/src/data/srdata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/src/data/srdata.py -------------------------------------------------------------------------------- /MindSpore version/src/edsr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/src/edsr_model.py -------------------------------------------------------------------------------- /MindSpore version/src/edsr_slim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/src/edsr_slim.py -------------------------------------------------------------------------------- /MindSpore version/src/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/src/metric.py -------------------------------------------------------------------------------- /MindSpore version/src/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/src/metrics.py -------------------------------------------------------------------------------- /MindSpore version/src/rcan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/src/rcan_model.py -------------------------------------------------------------------------------- /MindSpore version/src/vgg_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/src/vgg_model.py -------------------------------------------------------------------------------- /MindSpore version/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/train.py -------------------------------------------------------------------------------- /MindSpore version/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MindSpore version/utils/var_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/MindSpore version/utils/var_init.py -------------------------------------------------------------------------------- /PyTorch version/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/README.md -------------------------------------------------------------------------------- /PyTorch version/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/data/__init__.py -------------------------------------------------------------------------------- /PyTorch version/data/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/data/benchmark.py -------------------------------------------------------------------------------- /PyTorch version/data/bsd500.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/data/bsd500.py -------------------------------------------------------------------------------- /PyTorch version/data/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/data/common.py -------------------------------------------------------------------------------- /PyTorch version/data/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/data/demo.py -------------------------------------------------------------------------------- /PyTorch version/data/div2k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/data/div2k.py -------------------------------------------------------------------------------- /PyTorch version/data/div2kjpeg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/data/div2kjpeg.py -------------------------------------------------------------------------------- /PyTorch version/data/sr291.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/data/sr291.py -------------------------------------------------------------------------------- /PyTorch version/data/srdata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/data/srdata.py -------------------------------------------------------------------------------- /PyTorch version/data/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/data/video.py -------------------------------------------------------------------------------- /PyTorch version/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/dataloader.py -------------------------------------------------------------------------------- /PyTorch version/loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch version/loss/adversarial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/loss/adversarial.py -------------------------------------------------------------------------------- /PyTorch version/loss/contrast_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/loss/contrast_loss.py -------------------------------------------------------------------------------- /PyTorch version/loss/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/loss/discriminator.py -------------------------------------------------------------------------------- /PyTorch version/loss/perceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/loss/perceptual.py -------------------------------------------------------------------------------- /PyTorch version/loss/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/loss/vgg.py -------------------------------------------------------------------------------- /PyTorch version/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/main.py -------------------------------------------------------------------------------- /PyTorch version/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch version/model/carn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/model/carn.py -------------------------------------------------------------------------------- /PyTorch version/model/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/model/common.py -------------------------------------------------------------------------------- /PyTorch version/model/edsr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/model/edsr.py -------------------------------------------------------------------------------- /PyTorch version/model/rcan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/model/rcan.py -------------------------------------------------------------------------------- /PyTorch version/option.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/option.py -------------------------------------------------------------------------------- /PyTorch version/trainer/slim_contrast_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/trainer/slim_contrast_trainer.py -------------------------------------------------------------------------------- /PyTorch version/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch version/utils/niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/utils/niqe.py -------------------------------------------------------------------------------- /PyTorch version/utils/niqe_image_params.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/utils/niqe_image_params.mat -------------------------------------------------------------------------------- /PyTorch version/utils/spatial_trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/utils/spatial_trans.py -------------------------------------------------------------------------------- /PyTorch version/utils/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/utils/ssim.py -------------------------------------------------------------------------------- /PyTorch version/utils/utility.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/PyTorch version/utils/utility.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/README.md -------------------------------------------------------------------------------- /images/debug.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/images/debug.log -------------------------------------------------------------------------------- /images/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/images/model.png -------------------------------------------------------------------------------- /images/psnr-speed.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/images/psnr-speed.png -------------------------------------------------------------------------------- /images/psnr-tradeoff.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/images/psnr-tradeoff.png -------------------------------------------------------------------------------- /images/table.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/images/table.png -------------------------------------------------------------------------------- /images/tradeoff.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/images/tradeoff.png -------------------------------------------------------------------------------- /images/visual.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Booooooooooo/CSD/HEAD/images/visual.png --------------------------------------------------------------------------------