PyTorch transforms.Compose 图像变换组合详解

transforms.Compose 是 PyTorch 中的一个类，用于将多个图像变换组合在一起。其作用是将多个图像变换按照顺序组合起来，形成一个新的变换，可以直接应用于数据集中的图像数据上。

例如，如果我们想要对图像数据进行一系列的变换，比如先将图像缩放到指定的大小，然后将图像随机旋转一定角度，最后将图像转换为张量形式，可以使用 transforms.Compose 将这些变换组合起来，形成一个新的变换：

transform = transforms.Compose([
    transforms.Resize(256),
    transforms.RandomRotation(30),
    transforms.ToTensor()
])

上述代码中，我们将三个变换组合起来，分别是将图像缩放到 256x256 大小，随机旋转 0-30 度，最后将图像转换为张量形式。这样，我们可以直接将这个变换应用于数据集中的图像数据上。

总之，transforms.Compose 是一个很方便的工具，可以将不同的图像变换组合在一起，实现更加复杂的数据预处理操作。