Shortcuts

高斯噪声

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

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

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

输入张量也应为浮点类型,范围在[0, 1]之间。 此转换不支持PIL图像。

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

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

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