torchvision.transforms如何改变图像的大小并把大小统一
时间: 2024-03-27 21:39:11 浏览: 60
关于PyTorch源码解读之torchvision.models
5星 · 资源好评率100%
可以使用transforms.Resize()函数来改变图像的大小,并使用transforms.CenterCrop()函数将图像裁剪为相同的大小。以下是示例代码:
```python
import torchvision.transforms as transforms
# 定义变换
transform = transforms.Compose([
transforms.Resize(256), # 将图像调整为 256x256
transforms.CenterCrop(224), # 将图像裁剪为 224x224
transforms.ToTensor() # 将图像转换为张量
])
# 对图像进行变换
image = transform(image)
```
在上述代码中,首先使用transforms.Resize()函数将图像调整为256x256的大小,然后使用transforms.CenterCrop()函数将图像裁剪为224x224的大小,最后使用transforms.ToTensor()函数将图像转换为张量。这样,所有的图像都会被统一调整为相同的大小。
阅读全文