回头再详述,这里先上代码
import scipy.misc
import tensorflow as tf
import os
def crop_center(img,cropx,cropy):#待裁剪的图像,裁剪的宽度,裁剪的高度
y,x,_ = img.shape
startx = x//2-(cropx//2)
starty = y//2-(cropy//2)
return img[starty:starty+cropy,startx:startx+cropx]#返回裁剪后的图像
#获取训练集
def down_sample(in_dir,out1_dir,out2_dir,original_size,shrunk_size):#批量大小,目标尺寸,缩小尺寸
img_files = os.listdir(in_dir) #读取该路径下的文件名
img_size = len(img_files) #图片张数
for i in range(img_size):
img = scipy.misc.imread(in_dir+"/"+img_files[i])#读取图片
img = crop_center(img,original_size,original_size)#裁剪图片为第一张图片大小
x_img = scipy.misc.imresize(img,(shrunk_size,shrunk_size))#均衡缩放图片修改尺寸后
k = str(i)
scipy.misc.imsave(out1_dir+"/output_"+ k +".jpg",img)
scipy.misc.imsave(out2_dir+"/output_"+ k +".jpg",x_img)
return
down_sample('/home/zcm/tensorf/LR_SR/General-100','/home/zcm/tensorf/LR_SR/original','/home/zcm/tensorf/LR_SR/downsample',200,100)