torchvision.transforms.resize()用法
时间: 2024-05-14 20:12:09 浏览: 19
`torchvision.transforms.resize()` 函数用于调整图像的大小。
函数定义如下:
```python
torchvision.transforms.Resize(size, interpolation=2)
```
参数说明:
- `size`:所需的图像大小。可以是一个整数,表示将图像的较小边缩放到该大小,也可以是一个元组,表示缩放后的宽度和高度。
- `interpolation`:插值方法,可选值为 `PIL.Image.NEAREST`、`PIL.Image.BILINEAR`、`PIL.Image.BICUBIC` 或 `PIL.Image.LANCZOS`。默认值为 `PIL.Image.BILINEAR`。
示例代码:
```python
import torchvision.transforms as transforms
from PIL import Image
# 加载图像
img = Image.open('image.jpg')
# 调整图像大小
resize = transforms.Resize((224, 224))
img_resized = resize(img)
# 显示调整后的图像
img_resized.show()
```
在上述示例中,我们使用 `PIL` 库中的 `Image` 类加载了一张图像,并使用 `torchvision.transforms.Resize()` 函数将其缩放到了 (224, 224) 的大小。最后,我们使用 `show()` 方法显示了调整大小后的图像。
相关问题
torchvision.transforms.Resize() 参数
torchvision.transforms.Resize() 函数的参数是一个整数或者一个元组。整数表示将图像的短边缩放到指定大小,长边将按比例缩放。元组表示将图像的尺寸缩放到指定大小。例如:
- 如果参数是一个整数 n,那么图像将被缩放为短边为 n,长边按比例缩放。
- 如果参数是一个元组 (h, w),那么图像将被缩放为尺寸为 (h, w)。
示例用法:
```python
import torchvision.transforms as transforms
transform = transforms.Resize(256) # 将短边缩放为256,长边按比例缩放
transform = transforms.Resize((256, 256)) # 将图像尺寸缩放为 256x256
```
注意:在使用 torchvision.transforms.Resize() 函数时,可以将其作为数据预处理的一部分,用于调整输入图像的大小。
torchvision.transforms.Resize() 参数
torchvision.transforms.Resize()函数的参数有以下几个:
- size:要调整到的目标尺寸。可以是一个整数,表示将较小的边调整为该大小,或者是一个元组 (height, width),表示调整为给定的高度和宽度。
- interpolation:插值方法。可以是以下几种选项之一:
- PIL.Image.NEAREST:最近邻插值。
- PIL.Image.BILINEAR:双线性插值。
- PIL.Image.BICUBIC:双三次插值。
- PIL.Image.LANCZOS:Lanczos插值。
默认值为PIL.Image.BILINEAR。
示例用法:
```python
from torchvision import transforms
resize = transforms.Resize(size=(224, 224), interpolation=Image.BICUBIC)
```
这个例子展示了如何将图像调整为大小为224×224像素的尺寸,并使用双三次插值进行插值。