Compose¶
- class torchvision.transforms.v2.Compose(transforms: Sequence[Callable])[source]¶
将多个转换组合在一起。
此转换不支持 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 的示例