Compose¶
- class torchvision.transforms.v2.Compose(transforms: Sequence[Callable])[源代码]¶
将多个转换组合在一起。
此转换不支持 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
的示例开始使用转换 v2转换 v2:端到端目标检测/分割示例如何使用 CutMix 和 MixUp如何编写自己的 v2 转换Torchscript 支持