RandomChoice

class torchvision.transforms.v2.RandomChoice(transforms: Sequence[Callable], p: Optional[list[float]] = None)

從列表中隨機選擇一個變換並應用。

此轉換不支援 torchscript。

引數:

• transforms (sequence 或 torch.nn.Module) – 轉換列表

• p (list of python:floats 或 None, optional) – 每個變換被選中的機率。如果 p 的總和不為 1，它將被自動歸一化。如果 None (預設)，所有變換的機率都相同。

使用 RandomChoice 的示例

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

forward(*inputs: Any) → Any

不要覆蓋此方法！請使用 transform() 代替。