RandomApply¶
- class torchaug.transforms.RandomApply(transforms, p=0.5)[source]¶
Apply randomly a list of transformations with a given probability.
Note
In order to script the transformation, please use
torch.nn.ModuleListas input instead of list/tuple of transforms as shown below:>>> transforms = transforms.RandomApply(torch.nn.ModuleList([ >>> transforms.ColorJitter(), >>> ]), p=0.3) >>> scripted_transforms = torch.jit.script(transforms)
Make sure to use only scriptable transformations, i.e. that work with
torch.Tensor, does not require lambda functions orPIL.Image.- Parameters:
transforms (
Sequence[Union[Callable,ModuleList]]) – list of transformationsp (
float, optional) – probability of applying the list of transforms. Default:0.5