testing gaussian noise

test_gaussian_noise.py

# made by @mori_inj

import numpy as np
import matplotlib.pyplot as plt
import cv2
p = 2#np.inf
pert = 100#1000
im = cv2.imread('00000000.png')
#print(type(im))  # 처음부터 numpy
#print(im.shape)  # (H, W, C)
print(im.size)  # H x W x C
noise = np.random.normal(0, 1, np.size(im))
noise = noise / np.linalg.norm(noise, p) * pert
print(np.linalg.norm(noise, p), pert)
im += noise.reshape(im.shape).astype(np.uint8)
print(np.amax(im), np.amin(im), type(im[0, 0, 0]))  # np.uint8 (0 ~ 255)
plt.imshow(im[:,:,::-1], interpolation='bicubic')  # R G B
plt.show()