如何用OpenCV在Python中添加噪声（高斯/盐和胡椒等）

Question

我想知道在Python中是否有OpenCV或任何其他Python图像处理库中的函数，这些函数会给图像添加高斯或椒盐噪声？例如，在MATLAB中存在做同样工作的正向函数。

或者，如何使用带有OpenCV的Python为图像添加噪声？

Answer 1

我不知道PythonAPI中有什么方法，但是你可以用这个简单的代码给图像添加椒盐噪声。

import numpy as np
import random
import cv2

def sp_noise(image,prob):
    '''
    Add salt and pepper noise to image
    prob: Probability of the noise
    '''
    output = np.zeros(image.shape,np.uint8)
    thres = 1 - prob 
    for i in range(image.shape[0]):
        for j in range(image.shape[1]):
            rdn = random.random()
            if rdn < prob:
                output[i][j] = 0
            elif rdn > thres:
                output[i][j] = 255
            else:
                output[i][j] = image[i][j]
    return output

image = cv2.imread('image.jpg',0) # Only for grayscale image
noise_img = sp_noise(image,0.05)
cv2.imwrite('sp_noise.jpg', noise_img)