Compose — Torchvision 0.22 documentation (original) (raw)

class torchvision.transforms.v2.Compose(transforms: Sequence[Callable])[source]¶

Composes several transforms together.

This transform does not support torchscript. Please, see the note below.

Parameters:

transforms (list of Transform objects) – list of transforms to compose.

Example

transforms.Compose([ transforms.CenterCrop(10), transforms.PILToTensor(), transforms.ConvertImageDtype(torch.float), ])

Note

In order to script the transformations, please use torch.nn.Sequential as below.

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)

Make sure to use only scriptable transformations, i.e. that work with torch.Tensor, does not requirelambda functions or PIL.Image.

Examples using Compose:

Return the extra representation of the module.

To print customized extra information, you should re-implement this method in your own modules. Both single-line and multi-line strings are acceptable.

forward(*inputs: Any) → Any[source]¶

Do not override this! Use transform() instead.