数字图像处理 加性高斯白噪声

import numpy as np
import matplotlib.pyplot as plt
import cv2

def convert_2d(r):
    s = r + np.random.normal(0, 64, r.shape)
    if np.min(s) >= 0 and np.max(s) <= 255:
        return s;
    s = s - np.full(s.shape, np.min(s))
    s = s * 255 / np.max(s)
    s = s.astype(np.uint8)
    return s


def convert_3d(r):
    s_dsplit = []
    for d in range(r.shape[2]):
        rr = r[:, :, d]
        ss = convert_2d(rr)
        s_dsplit.append(ss)
    s = np.dstack(s_dsplit)
    return s


im = cv2.imread('tetet.jpg')
im_converted_mat = convert_3d(im)
cv2.imshow('addNoise', im_converted_mat)
cv2.imwrite('noise.jpg', im_converted_mat)
cv2.waitKey()