transforms.Compose
时间: 2023-11-01 22:56:07 浏览: 125
transforms.Compose是PyTorch中的一个类,用于将多个图像变换操作组合在一起。它接受一个包含多个变换操作的列表,并按顺序应用这些操作。
例如,假设我们有两个变换操作:Resize和ToTensor。我们可以使用transforms.Compose将它们组合在一起,使得我们可以在应用到图像上时一次性执行这两个操作。
以下是一个使用transforms.Compose的示例代码:
```python
import torch
from torchvision import transforms
# 定义两个变换操作
transform_resize = transforms.Resize((256, 256))
transform_to_tensor = transforms.ToTensor()
# 创建Compose对象并将变换操作组合在一起
composed_transform = transforms.Compose([transform_resize, transform_to_tensor])
# 应用组合的变换操作到图像上
image = torch.randn(3, 300, 300) # 随机生成一个形状为(3, 300, 300)的图像
transformed_image = composed_transform(image)
```
在上面的示例中,我们首先定义了两个变换操作:Resize和ToTensor。然后,我们使用transforms.Compose创建了一个Compose对象,并将这两个变换操作组合在一起。最后,我们将这个组合的变换操作应用到一个随机生成的图像上,得到一个经过变换的图像。
阅读全文