├── .gitattributes ├── README.md ├── dataset ├── dataset_denoise.py └── dataset_motiondeblur.py ├── evaluate.py ├── figs ├── method.png ├── test.png └── train.png ├── generate_patches_SIDD.py ├── losses.py ├── model.py ├── options.py ├── script ├── test.sh ├── train_denoise.sh ├── train_denoise_half.sh └── train_motiondeblur.sh ├── shuffleformer.py ├── test.py ├── test ├── eval_gopro.m ├── eval_realblur.py ├── eval_sidd.m ├── eval_spa.m ├── test.py ├── test_deblur.py ├── test_derain.py ├── test_dnd.py ├── test_gopro_hide.py ├── test_noise.py ├── test_realblur.py └── test_sidd.py ├── test_sidd.py ├── train ├── train_denoise.py ├── train_derain.py └── train_motiondeblur.py ├── utils ├── __init__.py ├── antialias.py ├── bundle_submissions.py ├── caculate_psnr_ssim.py ├── dataset_utils.py ├── dir_utils.py ├── image_utils.py ├── loader.py └── model_utils.py └── warmup_scheduler ├── __init__.py ├── run.py └── scheduler.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/.gitattributes -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/README.md -------------------------------------------------------------------------------- /dataset/dataset_denoise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/dataset/dataset_denoise.py -------------------------------------------------------------------------------- /dataset/dataset_motiondeblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/dataset/dataset_motiondeblur.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/evaluate.py -------------------------------------------------------------------------------- /figs/method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/figs/method.png -------------------------------------------------------------------------------- /figs/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/figs/test.png -------------------------------------------------------------------------------- /figs/train.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/figs/train.png -------------------------------------------------------------------------------- /generate_patches_SIDD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/generate_patches_SIDD.py -------------------------------------------------------------------------------- /losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/losses.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/model.py -------------------------------------------------------------------------------- /options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/options.py -------------------------------------------------------------------------------- /script/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/script/test.sh -------------------------------------------------------------------------------- /script/train_denoise.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/script/train_denoise.sh -------------------------------------------------------------------------------- /script/train_denoise_half.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/script/train_denoise_half.sh -------------------------------------------------------------------------------- /script/train_motiondeblur.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/script/train_motiondeblur.sh -------------------------------------------------------------------------------- /shuffleformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/shuffleformer.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/test.py -------------------------------------------------------------------------------- /test/eval_gopro.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/test/eval_gopro.m -------------------------------------------------------------------------------- /test/eval_realblur.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/eval_sidd.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/test/eval_sidd.m -------------------------------------------------------------------------------- /test/eval_spa.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/test/eval_spa.m -------------------------------------------------------------------------------- /test/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/test/test.py -------------------------------------------------------------------------------- /test/test_deblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/test/test_deblur.py -------------------------------------------------------------------------------- /test/test_derain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/test/test_derain.py -------------------------------------------------------------------------------- /test/test_dnd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/test/test_dnd.py -------------------------------------------------------------------------------- /test/test_gopro_hide.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/test/test_gopro_hide.py -------------------------------------------------------------------------------- /test/test_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/test/test_noise.py -------------------------------------------------------------------------------- /test/test_realblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/test/test_realblur.py -------------------------------------------------------------------------------- /test/test_sidd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/test/test_sidd.py -------------------------------------------------------------------------------- /test_sidd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/test_sidd.py -------------------------------------------------------------------------------- /train/train_denoise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/train/train_denoise.py -------------------------------------------------------------------------------- /train/train_derain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/train/train_derain.py -------------------------------------------------------------------------------- /train/train_motiondeblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/train/train_motiondeblur.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/antialias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/utils/antialias.py -------------------------------------------------------------------------------- /utils/bundle_submissions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/utils/bundle_submissions.py -------------------------------------------------------------------------------- /utils/caculate_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/utils/caculate_psnr_ssim.py -------------------------------------------------------------------------------- /utils/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/utils/dataset_utils.py -------------------------------------------------------------------------------- /utils/dir_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/utils/dir_utils.py -------------------------------------------------------------------------------- /utils/image_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/utils/image_utils.py -------------------------------------------------------------------------------- /utils/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/utils/loader.py -------------------------------------------------------------------------------- /utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/utils/model_utils.py -------------------------------------------------------------------------------- /warmup_scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/warmup_scheduler/__init__.py -------------------------------------------------------------------------------- /warmup_scheduler/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/warmup_scheduler/run.py -------------------------------------------------------------------------------- /warmup_scheduler/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiexiaou/ShuffleFormer/HEAD/warmup_scheduler/scheduler.py --------------------------------------------------------------------------------