Compose¶
- class torchvision.transforms.Compose(transforms)[源]¶
將多個轉換(transform)組合在一起。此轉換不支援 torchscript。請參閱下面的注意事項。
- 引數:
transforms (list of
Transformobjects) – 要組合的轉換列表。
示例
>>> 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的示例