RandomChoice

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

随机从列表中选择单个变换进行应用。

此转换不支持 torchscript。

参数:

• transforms (sequence 或 torch.nn.Module) – 变换列表

• p (python:floats 列表 或 None, 可选) – 每个变换被选中的概率。如果 p 的总和不为 1，则会自动进行归一化。如果为 None (默认)，所有变换的概率都相同。

使用 RandomChoice 的示例

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

forward(*inputs: Any) → Any

不要覆盖此方法！请使用 transform() 代替。