RandomApply¶
- class torchvision.transforms.RandomApply(transforms, p=0.5)[source]¶
以给定概率随机应用一个变换列表。
注意
为了能够脚本化(script)此变换,请使用
torch.nn.ModuleList
作为输入,而非列表/元组形式的变换,如下所示>>> transforms = transforms.RandomApply(torch.nn.ModuleList([ >>> transforms.ColorJitter(), >>> ]), p=0.3) >>> scripted_transforms = torch.jit.script(transforms)
确保仅使用可脚本化(scriptable)的变换,即那些与
torch.Tensor
配合使用、不需要 lambda 函数或PIL.Image
的变换。- 参数:
transforms (sequence 或 torch.nn.Module) – 变换列表
p (float) – 概率
使用
RandomApply
的示例