样本预处理
以500*332的图为例
import torchvision.transforms as transforms from PIL import Image img = Image.open('D:/cat.jpg') #缩放成高宽都是256的正方形(小变大的话,是插值不会填充黑边) resize = transforms.Resize((256,256)) img_resized = resize(img) img_resized.show() #中心裁剪 cenCrop = transforms.CenterCrop(224) img_cenCrop = cenCrop(img_resized) img_cenCrop.show()
注意 resize = transforms.Resize(256) 这种写法,是较短的边缩放到256,较长的边等比例缩放(即长边>256)
一般推荐 所有的尺寸输入要用括号!!