快捷方式

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 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

查找开发资源并获得问题解答

查看资源