transforms.Resize
时间: 2024-05-06 18:17:58 浏览: 155
resize
transforms.Resize 是 PyTorch 中的一个数据预处理函数,用于调整输入数据的大小。该函数可以接收一个整数或一个元组作为参数,分别表示调整后的图片大小。如果传入一个整数,则表示将输入数据的短边缩放到指定大小,长边按比例缩放。如果传入一个元组,则表示将输入数据的大小调整为元组中指定的大小。
示例:
```python
from torchvision.transforms import transforms
transform = transforms.Compose([
transforms.Resize(256)
])
# 对输入数据进行预处理
input_data = transform(input_data)
```
上述示例中,Resize 函数将输入数据的短边缩放到 256,长边按比例缩放。
阅读全文