RandomApply¶
- class torchvision.transforms.v2.RandomApply(transforms: Union[Sequence[Callable], ModuleList], p: float = 0.5)[源代码]¶
以给定的概率随机应用一系列变换。
注意
为了对转换进行脚本化,请使用
torch.nn.ModuleList
作为输入,而不是列表/元组的转换,如下所示:>>> transforms = transforms.RandomApply(torch.nn.ModuleList([ >>> transforms.ColorJitter(), >>> ]), p=0.3) >>> scripted_transforms = torch.jit.script(transforms)
请确保只使用可脚本化的转换,即可以与
torch.Tensor
一起工作的转换,不需要 lambda 函数或PIL.Image
。- 参数:
transforms (sequence 或 torch.nn.Module) – 转换列表
p (float) – 应用转换列表的概率
RandomApply
的用法示例