空域噪声滤波器

import os
import numpy as np
import cv2
import matplotlib.pyplot as plt
plt.rcParams['font.sans-serif']=['SimHei']
plt.rcParams['axes.unicode_minus']=False
def put(path):
    img=cv2.imread(path,1)
    img=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
    rows,cols=img.shape[:2]
    #添加噪声
    img_noise=img.copy()
    for i in range(5000):
        x=np.random.randint(0,rows)
        y=np.random.randint(0,cols)
        img_noise[x,y]=img[x,y]+np.random.randint(50,100)
    #均值滤波
    res1=cv2.blur(img_noise,(9,9))
    #统计排序滤波器
    #中值滤波
    res2=cv2.medianBlur(img_noise,5)
    plt.subplot(2,2,1),plt.axis('off'),plt.imshow(img,plt.cm.gray),plt.title('原始图像')
    plt.subplot(2,2,2),plt.axis('off'),plt.imshow(img_noise,plt.cm.gray),plt.title('加入噪声后图像')
    plt.subplot(2,2,3),plt.axis('off'),plt.imshow(res1,plt.cm.gray),plt.title('均值滤波器')
    plt.subplot(2,2,4),plt.axis('off'),plt.imshow(res2,plt.cm.gray),plt.title('中值滤波器')
    plt.savefig('C:/Users/xpp/Desktop/result.png')
    plt.show()
put(r'C:/Users/xpp/Desktop/Lena.png')