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 的示例：

變換的插圖