Pytorch transforms.Resize()的简单用法

本文介绍了在使用Python的PIL库和torchvision库处理图像时,尺寸操作的细节。通过示例代码展示了PIL.Image对象的size属性返回的图像尺寸顺序与torchvision.transforms.Resize方法参数顺序的不同,可能导致图像尺寸反转的问题。修复这一问题的方法是调整Resize的参数顺序,以确保正确地保存处理后的图像尺寸。
摘要由CSDN通过智能技术生成

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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值