快捷方式

Compose

class torchvision.transforms.Compose(transforms)[源]

將多個轉換(transform)組合在一起。此轉換不支援 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 入門

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源