RandomChoice

class torchaug.transforms.RandomChoice(transforms, p=None)[source]

Apply single transformation randomly picked from a list.

This transform does not support torchscript.

Parameters:
  • transforms (Sequence[Callable]) – list of transformations

  • p (Optional[List[float]], optional) – probability of each transform being picked. If p doesn’t sum to 1, it is automatically normalized. If None (default), all transforms have the same probability. Default: None

forward(*inputs)[source]

Performs forward pass of the transform.

Parameters:

inputs (Any) – Inputs to the transform.

Return type:

Any

Returns:

Transformed inputs.