代码实例:
num_val_images = len(listdir(val_hr_path))
for i in tqdm(range(num_val_images)):
img_name = listdir(val_hr_path)[i]
img = io.imread('{}{}'.format(val_hr_path, img_name))
scaled_image = rescale(img, (1/scale), channel_axis=-1, anti_aliasing=1)
io.imsave('{}{}'.format(val_lr_path, img_name), scaled_image)
运行后出现TypeError: Cannot handle this data type: (1, 1, 3), <f8
原因:
图片数据类型与要导入的数据类型不符。
解决方案:
在第6行下添加scaled_image=(scaled_image*255).astype(np.uint8)