Compose

class torchvision.transforms.v2.Compose(transforms: Sequence[Callable])

将多个变换组合在一起。

此变换不支持 torchscript。请参见下面的说明。

参数:

transforms (list of Transform objects) – 要组合的变换列表。

示例

>>> 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 的示例

Torchscript 支持

Torchscript 支持

如何编写自己的 v2 变换

如何编写自己的 v2 变换

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

旋转边界框上的变换

旋转边界框上的变换

变换 v2：端到端目标检测/分割示例

变换 v2：端到端目标检测/分割示例

变换 v2 入门

变换 v2 入门

extra_repr() → str

返回模块的额外表示。

要打印自定义额外信息，您应该在自己的模块中重新实现此方法。单行和多行字符串均可接受。

forward(*inputs: Any) → Any

不要覆盖此方法！请使用 transform() 代替。