Python PyTorch Compose用法及代碼示例

本文簡要介紹python語言中 torchvision.transforms.Compose 的用法。

用法:

class torchvision.transforms.Compose(transforms)

參數：

transforms(列表Transform對象) -要組合的變換列表。

將多個變換組合在一起。此轉換不支持 torchscript。請參閱下麵的注釋。

示例

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

視頻 API