Python PyTorch RandomApply用法及代码示例

本文简要介绍python语言中 torchvision.transforms.RandomApply 的用法。

用法:

class torchvision.transforms.RandomApply(transforms, p=0.5)

参数：

• transforms(序列或者torch.nn.Module) -转换列表

• p(float) -可能性

以给定的概率随机应用一系列变换。

注意

为了编写转换脚本，请使用 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 。

使用 RandomApply 的示例：

变换的插图