Compose

class torchvision.transforms.Compose(transforms)

將多個轉換組合在一起。此轉換不支援 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 的示例

如何編寫自己的 v2 變換

如何編寫自己的 v2 變換

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

旋轉邊界框上的變換

旋轉邊界框上的變換

變換 v2：端到端目標檢測/分割示例

變換 v2：端到端目標檢測/分割示例

變換 v2 入門

變換 v2 入門