Pytorch transforms.Resize()的简单用法

PILImage对象size属性返回的是w, h，而resize的参数顺序是h, w

from PIL import Image
from torchvision import transforms

img = Image.open('1.jpg')
w, h = img.size # 1.jpg 的尺寸为 w×h
resize = transforms.Resize([224,244])
img = resize(img)
img.save('2.jpg') # 2.jpg 的尺寸为 244×224
resize2 = transforms.Resize([h, w])
img = resize2(img)
img.save('3.jpg') # 3.jpg 的尺寸为 w×h

• 转载博客：Pytorch transforms.Resize()的简单用法