样本预处理

以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)

一般推荐 所有的尺寸输入要用括号!!

 

posted @ 2024-10-24 16:40  夕西行  阅读(4)  评论(0编辑  收藏  举报