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()