1. 添加高斯噪声

import cv2
import numpy as np

def imnoise_gasussian(image, mean=0, var=0.001):
    ''' 添加高斯噪声
        mean : 均值 
        var : 方差
    '''

    image = np.array(image/255, dtype=float)
    noise = np.random.normal(mean, var ** 0.5, image.shape)
    img_noise = image + noise

    img_noise = np.clip(img_noise, 0, 1.0)
    img_noise = np.uint8(img_noise*255)

    return img_noise


if __name__ == '__main__':
    img_file = './monarch.png'
    img = cv2.imread(img_file, cv2.IMREAD_COLOR)

    img_noise = imnoise_gasussian(img, 0, 0.01)

    cv2.imshow('noise', img_noise)
    cv2.waitKey(0)
    cv2.destroyAllWindows()

高斯噪声，mean=0，var=0.01

2. 添加椒盐噪声

import cv2
import numpy as np
import random

def imnoise_sp(image, prob):
    '''  添加椒盐噪声
        prob:噪声比例 
    '''
    img_noise = np.copy(image)
    thres = 1 - prob 
    for i in range(image.shape[0]):
        for j in range(image.shape[1]):
            rdn = random.random()
            if rdn < prob:
                img_noise[i][j] = 0
            elif rdn > thres:
                img_noise[i][j] = 255

    return img_noise

if __name__ == '__main__':
    img_file = './monarch.png'
    img = cv2.imread(img_file, cv2.IMREAD_COLOR)

    img_noise = imnoise_sp(img, 0.01)

    cv2.imshow('noise', img_noise)
    cv2.waitKey(0)
    cv2.destroyAllWindows()

添加椒盐噪声，prob=0.01