GaussianNoise

class torchvision.transforms.v2.GaussianNoise(mean: float = 0.0, sigma: float = 0.1, clip=True)

向图像或视频添加高斯噪声。

输入张量应为 […, 1 或 3, H, W] 格式，其中 … 表示它可以具有任意数量的前导维度。批次中的每个图像或帧将独立转换，即添加到每个图像的噪声将不同。

输入张量也应为 [0, 1] 范围内的浮点 dtype。此转换不支持 PIL 图像。

参数:

• mean (float) – 采样正态分布的均值。默认为 0。

• sigma (float) – 采样正态分布的标准差。默认为 0.1。

• clip (bool, 可选) – 是否在添加噪声后将值裁剪到 [0, 1] 范围内。默认为 True。

transform(inpt: Any, params: Dict[str, Any]) → Any

覆盖自定义转换的方法。

请参阅 如何编写自己的 v2 转换