Compose

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

將多個變換組合在一起。

此變換不支援 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 的示例

Torchscript 支援

Torchscript 支援

如何編寫自己的 v2 變換

如何編寫自己的 v2 變換

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

Transforms v2: 端到端目標檢測/分割示例

Transforms v2: 端到端目標檢測/分割示例

Transforms v2 入門

Transforms v2 入門

extra_repr() → str

返回模組的額外表示。

要列印自定義的額外資訊，您應該在自己的模組中重新實現此方法。單行和多行字串均可接受。

forward(*inputs

:

Any) → Any

請勿覆蓋此方法！請改用 transform()。