torchvision.transforms.Compose
时间: 2023-12-28 18:05:43 浏览: 80
torchvision.transforms.Compose 是一个用于组合多个图像转换操作的函数,它将这些操作串联起来以便一次性地应用到图像上。
例如,假设我们有两个图像转换操作:Resize 和 ToTensor。我们可以使用 Compose 函数将它们组合在一起,如下所示:
```python
from torchvision import transforms
transform = transforms.Compose([
transforms.Resize((256, 256)),
transforms.ToTensor()
])
```
在上述代码中,我们创建了一个 transform 对象,该对象是由两个操作组成的 Compose 对象。第一个操作是 Resize,它将图像的大小调整为 (256, 256)。第二个操作是 ToTensor,它将图像转换为 Tensor 格式。
然后,我们可以将这个 transform 对象应用到图像上,如下所示:
```python
import torch
from PIL import Image
image = Image.open('image.jpg')
transformed_image = transform(image)
```
在上述代码中,我们打开了一张图像,并将 transform 对象应用到图像上,得到了经过转换后的图像 transformed_image。
这样,我们就可以使用 torchvision.transforms.Compose 来方便地组合多个图像转换操作,并将其应用到图像上。
阅读全文