给图像增加高斯噪声

Question

我正在尝试使用以下代码向一些图像添加高斯噪声

import numpy as np
import cv2
import glob 
mean = 0
var = 10
sigma = var ** 0.5
gaussian = np.random.normal(mean, sigma, (224, 224)) 



for image in glob.glob('/home/aub/myflower/flower_photos/dandelion/*.jpg'):
    img = cv2.imread(image)
    noisy_image = np.zeros(img.shape, np.float32)

    if len(img.shape) == 2:
        noisy_image = img + gaussian
    else:
        noisy_image[:, :, 0] = img[:, :, 0] + gaussian
        noisy_image[:, :, 1] = img[:, :, 1] + gaussian
        noisy_image[:, :, 2] = img[:, :, 2] + gaussian

        cv2.normalize(noisy_image, noisy_image, 0, 255, cv2.NORM_MINMAX, dtype=-1)
noisy_image = noisy_image.astype(np.uint8)

       cv2.imshow("img", img)
       cv2.imshow("gaussian", gaussian)
       cv2.imshow("noisy", noisy_image)
cv2.waitKey(0)

但它不起作用，它给我以下错误

noisy_image [:,：，0] = img [:,：，0] +高斯ValueError：操作数不能与形状一起广播（315,500）（224,224）

请检查并提供反馈。

Answer 1

看起来像您的图像形状是(315,500)，而的形状gaussian是(224,224)。尝试将您的高斯初始化更改为

gaussian = np.random.normal(mean, sigma, (img.shape[0],img.shape[1])) 

顺便说一句：您可以替换这些行

noisy_image[:, :, 0] = img[:, :, 0] + gaussian
noisy_image[:, :, 1] = img[:, :, 1] + gaussian
noisy_image[:, :, 2] = img[:, :, 2] + gaussian

与

noisy_image = img + gaussian

效果相同：添加gaussian到每个频道。