Compose

class torchvision.transforms.Compose(transforms)

将多个变换组合在一起。此变换不支持 torchscript。请查看下面的注释。

参数:

transforms (Transform 对象列表) – 要组合的变换列表。

示例

>>> transforms.Compose([
>>>     transforms.CenterCrop(10),
>>>     transforms.PILToTensor(),
>>>     transforms.ConvertImageDtype(torch.float),
>>> ])

注意

为了编写变换脚本，请使用如下所示的 torch.nn.Sequential。

>>> transforms = torch.nn.Sequential(
>>>     transforms.CenterCrop(10),
>>>     transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225)),
>>> )
>>> scripted_transforms = torch.jit.script(transforms)

确保仅使用可编写脚本的变换，即可以使用 torch.Tensor，不需要 lambda 函数或 PIL.Image 的变换。

使用 Compose 的示例

transforms v2 入门

transforms v2 入门

Transforms v2：端到端对象检测/分割示例

Transforms v2：端到端对象检测/分割示例

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

如何编写您自己的 v2 变换

如何编写您自己的 v2 变换

视频 API

视频 API