torchvision.transforms中的Resize用法

torchvision.transforms.Resize 主要用于调整图像的尺寸,而不会改变数据内容或通道顺序。torchvision.transforms.Resize 可以对PIL或tensor进行处理。

用法1(处理PIL图像):

import torchvision.transforms as transforms
from PIL import Image

# 定义 Resize 变换
resize_transform = transforms.Resize((100, 100))  # 调整为 100x100
# 或者
resize_transform = transforms.Resize(100)  # 长边调整为 100,短边等比例缩放

# 读取图像
img = Image.open("example.jpg")  

# 应用 Resize 变换
resized_img = resize_transform(img)

用法2(处理tensor张量):

import torch
import torchvision.transforms as transforms

img_tensor = torch.randn(3, 200, 300)  # 假设是一个 RGB 图像
resize_transform = transforms.Resize((100, 150))  
resized_tensor = resize_transform(img_tensor)  # 变成 (3, 100, 150)

Resize的参数如下:

transforms.Resize(size, interpolation=InterpolationMode.BILINEAR, max_size=None, antialias=None)
  • size:目标尺寸。可以是单个值(长边缩放)或 (H, W)(强制缩放)。
  • interpolation:插值方法,如 BILINEAR(默认)、NEARESTBICUBICLANCZOS
  • max_size:size 是单个值时,限制最大尺寸(避免过度缩放)。
  • antialias:是否抗锯齿True/False,仅对 PIL.Image 有效)。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值