超分辨率中根據原圖生成測試集（低分辨圖或高分辨圖）

超分辨率生成訓練圖片代碼

 

 1 from PIL import Image
 2 import os
 3 import cv2
 4 def produceImage(file_in, width, height, file_out):
 5     image = Image.open(file_in)
 6     resized_image = image.resize((width, height), Image.ANTIALIAS)
 7 
 8     resized_image.save(file_out)
 9 
10 if __name__ == '__main__':
11 
12     #該路徑爲原圖路徑
13     path=r"H:\jianfeng\project\super_resolution\dataset\urban100_img\X3"
14     #該路徑爲保存路徑
15     save=r"H:\jianfeng\project\super_resolution\dataset\urban100_img\X3_HR"
16     fileList = os.listdir(path)
17 
18     for i in fileList:
19         a=os.path.join(path,i)
20 
21         pic = cv2.imread(a)
22 
23         width = pic.shape[1]
24         height = pic.shape[0]
25         #修改你要轉換的分辨率 這裏須要將分辨率乘以三倍，若是須要下降4倍則width//4，height//4
26         #遇到除不盡狀況，會自動規整，須要再作一次還原操做，
27         # 好比下降3倍後1024-->341 須要在將341圖片乘以三倍從新生成1023做爲HR圖
28         produceImage(a, width*3, height*3, os.path.join(save,i))