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 入門

轉換 v2：端到端物件偵測/分割範例

轉換 v2：端到端物件偵測/分割範例

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

如何編寫自己的 v2 轉換

如何編寫自己的 v2 轉換

Torchscript 支援

Torchscript 支援

extra_repr() → str

傳回模組的額外表示。

要列印自訂的額外資訊，您應該在自己的模組中重新實作此方法。單行和多行字串都是可接受的。

forward(*inputs: Any) → Any

請勿覆寫此項！請改用 transform()。