快捷方式

随机应用

class torchvision.transforms.RandomApply(transforms, p=0.5)[source]

以给定概率随机应用一组变换。

注意

为了对变换进行脚本化,请使用 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 的示例

变换的说明

变换的说明
forward(img)[source]

定义每次调用时执行的计算。

应该由所有子类覆盖。

注意

虽然需要在这个函数中定义前向传递的配方,但应该随后调用 Module 实例,而不是这个,因为前者负责运行注册的钩子,而后者会静默忽略它们。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源