采用scripy.misc.imreshape()来修改尺寸

来源:互联网 发布:淘宝京东评价体系 编辑:程序博客网 时间:2024/06/14 09:15

回头再详述,这里先上代码

import scipy.miscimport tensorflow as tfimport osdef 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)    returndown_sample('/home/zcm/tensorf/LR_SR/General-100','/home/zcm/tensorf/LR_SR/original','/home/zcm/tensorf/LR_SR/downsample',200,100)
原创粉丝点击