终于通了,对数据路径进行处理的问题,
cd codes/data_scripts
python extract_subimages_test.py
python divide_sub_images_test.py
其中python extract_subimages_test.py是对数据进行预处理,预处理的方法是将X4目录下的图片转换为LR,并存在相应路径下,其中更改的部分为:
GT_folder = '/codes/data/DIV2K_valid_HR' # fix to your path
LR_folder =‘/codes/data/DIV2K_valid_LR_bicubic/X4' # fix to your path
save_GT_folder = '/codes/data/DIV2K_valid_HR_sub/GT'
save_LR_folder = '/codes/data/DIV2K_valid_HR_sub/LR'
最后怎得都出不来LR模块,将路径强制输出:
opt['input_folder'] = '/codes/data/DIV2K_valid_LR_bicubic/X4'
opt['save_folder'] = '/codes/data/DIV2K_valid_HR_sub/LR'
最终划分成功了,简单对RCAN模型进行一个测试;