GaussianNoise

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

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

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

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

参数:

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

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

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

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

用于覆盖自定义变换的方法。

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