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() 代替。