对图片加噪声

tensor 版本

from PIL import Image
import torch
from torchvision import transforms as T


def add_noise(input_img, noise_sigma):
    noise_sigma = noise_sigma / 255
    noise_img = torch.clamp(input_img+noise_sigma*torch.randn_like(input_img), 0.0, 1.0)

    return noise_img


img = Image.open('./testgt.png')
img = T.ToTensor()(img)
img_noise = add_noise(img, 20)
img_noise = T.ToPILImage()(img_noise)
img_noise.save('./test.png')

numpy版本

noise_image = image + np.random.randn(*image.shape) * 20