├── CAVE_1024_28 └── train_list.txt ├── Cal_para_flops.py ├── Checkpoint └── Null ├── Checkpoint_pretrain └── Null ├── Dataset.py ├── LICENSE ├── Label └── null ├── Model.py ├── README.md ├── Result └── null ├── Test.py ├── Test_data └── test_list.txt ├── Test_pretrain.py ├── Train.py ├── Utils.py ├── cal_psnr_ssim.m ├── cal_ssim.m └── mask_256_28.mat /CAVE_1024_28/train_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhangJC-2k/SSR/HEAD/CAVE_1024_28/train_list.txt -------------------------------------------------------------------------------- /Cal_para_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhangJC-2k/SSR/HEAD/Cal_para_flops.py -------------------------------------------------------------------------------- /Checkpoint/Null: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /Checkpoint_pretrain/Null: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhangJC-2k/SSR/HEAD/Dataset.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhangJC-2k/SSR/HEAD/LICENSE -------------------------------------------------------------------------------- /Label/null: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhangJC-2k/SSR/HEAD/Model.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhangJC-2k/SSR/HEAD/README.md -------------------------------------------------------------------------------- /Result/null: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /Test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhangJC-2k/SSR/HEAD/Test.py -------------------------------------------------------------------------------- /Test_data/test_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhangJC-2k/SSR/HEAD/Test_data/test_list.txt -------------------------------------------------------------------------------- /Test_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhangJC-2k/SSR/HEAD/Test_pretrain.py -------------------------------------------------------------------------------- /Train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhangJC-2k/SSR/HEAD/Train.py -------------------------------------------------------------------------------- /Utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhangJC-2k/SSR/HEAD/Utils.py -------------------------------------------------------------------------------- /cal_psnr_ssim.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhangJC-2k/SSR/HEAD/cal_psnr_ssim.m -------------------------------------------------------------------------------- /cal_ssim.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhangJC-2k/SSR/HEAD/cal_ssim.m -------------------------------------------------------------------------------- /mask_256_28.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhangJC-2k/SSR/HEAD/mask_256_28.mat --------------------------------------------------------------------------------