图像预处理,需要进行图像的crop,resize和重命名save。代码如下:
import os
from PIL import Image
'''train'''
path = '/Users/xuqiong/AgeGender/test_img_process/1_allface/test/' #表示需要命名处理的文件夹
pathnew = '/Users/xuqiong/AgeGender/test_img_process/2_allcrop/test/'
filelist = os.listdir(path) #获取文件路径
cbox = [0,0,0,0]
for item in filelist:
if item == '.DS_Store':
continue
imgpath = path+item
imgpathnew = pathnew + item
img = Image.open(imgpath)
h = img.height
w = img.width
if h > w:
delta = int((h-w)/2)
cbox = [0,delta,w,w]
else:
delta = int((w-h) / 2)
cbox = [delta,0,h,h]
image_cropped = img.crop(cbox)
image_cropped.save(imgpathnew)
#image_cropped = img.crop(cbox)
#image_resized = image_cropped.resize((64,64))
#image_resized.save(imgpathnew)
#image_resized = img.resize((64, 64))
#image_resized.save(imgpathnew)