如何在图像上添加高斯噪声

Question

我想创建一个函数，将高斯噪声添加到稍后将使用的单个输入。 我正在 Python 中使用 imageio 模块。

def gaussian_noise(x, var):

Answer 1

我认为 x 是要添加高斯噪声的张量，var 是高斯噪声的方差。

你的问题很模糊，但你可以像这样添加高斯噪声：

import torch

def gaussian_noise(x, var):
  return torch.normal(0, var, size=x.shape)

T = torch.ones(4, 5)
T += gaussian_noise(T, 0.1)
print(T)

Answer 2

有一个直接为此目的量身定制的函数https://pytorch.org/vision/stable/ generated/torchvision.transforms.v2.GaussianNoise.html#torchvision.transforms.v2.GaussianNoise