def transform(image, pre_height, pre_width, height, width, is_crop):
if is_crop:
new_image = center_crop(image, pre_height, pre_width, height, width)
else:
new_image = scipy.misc.imresize(image, [height, width])
return np.array(new_image)/127.5 - 1.
评论列表
文章目录