transforms.CenterCrop
时间: 2023-10-17 21:08:29 浏览: 177
S-transform
transforms.CenterCrop 是 PyTorch 中的一个图像预处理函数,用于对图像进行中心裁剪。具体来说,它会将给定的图像从中心开始裁剪,裁剪后保留的区域大小由给定参数决定。如果输入的图像尺寸小于给定参数,则不会进行任何操作。
使用示例:
```python
from torchvision import transforms
from PIL import Image
# 定义预处理函数
transform = transforms.Compose([
transforms.CenterCrop(224), # 中心裁剪为 224 x 224
transforms.ToTensor() # 转为张量
])
# 加载图像
img = Image.open("image.jpg")
# 预处理图像
img_processed = transform(img)
```
阅读全文