核心的思想: 对图片中的某个元素增加噪声,然后返回添加噪声以后的图片。

代码:

# 对图像增加高斯噪声,possion噪声
from numpy import *
from scipy import *
import numpy as np
import cv2
srcImage=cv2.imread("lena.jpg")
print(srcImage.shape)
cv2.namedWindow("origin image")
cv2.imshow("origin image",srcImage)
k=cv2.waitKey(0)

# 把原始的图像转化为灰度的图像
gray_image=cv2.cvtColor(srcImage,cv2.COLOR_BGR2GRAY)
print(gray_image.shape)
cv2.imshow("gray image",gray_image)
k=cv2.waitKey(0)

image=np.array(gray_image/255,dtype=float)
percent=0.01
num=int(percent*image.shape[0]*image.shape[1])
print(num)

for i in range(num):
    # 获得随机的宽和高
    temp1=np.random.randint(image.shape[0])
    temp2=np.random.randint(image.shape[1])
    mean=0
    var=0.04
    noise=np.random.normal(mean,var**0.5,1)
    image[temp1][temp2]+=noise
out=image
print(out)
if out.min()<0:
    low_clip=-1
else:
    low_clip=0
out=np.clip(out,low_clip,1)
gauss_image=np.array(out*255,dtype='uint8')
print(gauss_image.shape)
cv2.imshow("gauss image",gauss_image)
k=cv2.waitKey(0)


image=np.array(gray_image/255,dtype=float)
percent=0.001
num=int(percent*image.shape[0]*image.shape[1])
print(num)

for i in range(num):
    # 获得随机的宽和高
    temp1=np.random.randint(image.shape[0])
    temp2=np.random.randint(image.shape[1])
    scale=150
    noise=np.random.poisson(scale,1)
    image[temp1][temp2]+=noise
out=image
print(out)
if out.min()<0:
    low_clip=-1
else:
    low_clip=0
out=np.clip(out,low_clip,1)
possion_image=np.array(out*255,dtype='uint8')
print(possion_image.shape)
cv2.imshow("possion image",possion_image)
k=cv2.waitKey(0)

 

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐