pytorch resize图片
时间: 2023-04-03 14:03:40 浏览: 258
pytorch之Resize()函数具体使用详解
可以使用pytorch中的torchvision.transforms.Resize()函数来对图片进行resize操作。例如,将图片resize为224x224的大小,可以使用以下代码:
```
import torch
import torchvision.transforms as transforms
from PIL import Image
img = Image.open('image.jpg')
transform = transforms.Compose([
transforms.Resize((224, 224)),
transforms.ToTensor()
])
img = transform(img)
```
其中,transforms.Resize()函数接受一个tuple作为参数,表示resize后的大小。transforms.ToTensor()函数可以将PIL Image对象转换为torch.Tensor对象。
阅读全文