transforms.Compose
是 PyTorch 中 torchvision.transforms 模块提供的一个类,用于将多个图像转换操作组合成一个序列。这个类允许用户定义一个转换操作的列表,然后将这个列表中的所有操作按照顺序应用到图像数据上。
在 PyTorch 中,图像预处理通常需要进行多个步骤,例如将图像转换为张量、归一化、随机裁剪等。使用 transforms.Compose
可以方便地将这些操作按照指定的顺序组合起来,形成一个完整的预处理流程。
以下是一个使用 transforms.Compose
的示例:
from torchvision import transforms # 定义一系列图像转换操作 transform = transforms.Compose([ transforms.RandomResizedCrop(224), transforms.RandomHorizontalFlip(), transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)), ]) # 应用转换操作到图像数据 processed_image = transform(original_image)
在这个例子中,transforms.Compose
接受一个由图像转换操作组成的列表,然后通过 processed_image = transform(original_image)
将这些操作应用到原始图像上。最终得到的 processed_image
就是经过预处理后的图像数据。