中值滤波器

中值滤波器属于非线性滤波器，中值滤波是对整幅图像求解中位数的过程。

import cv2 as cv
import matplotlib.pyplot as plt
import math
import numpy as np
plt.rcParams['font.sans-serif'] =['SimHei']  #显示中文标签

def get_median(data):
    data.sort()
    half = len(data) // 2
    return data[half]
#计算灰度图像的中值滤波
def my_median_blur_gray(image, size):
    data = []
    sizepart = int(size/2)
    for i in range(image.shape[0]):
        for j in range(image.shape[1]):
            for ii in range(size):
                for jj in range(size):
                    #首先判断所以是否超出范围，也可以事先对图像进行零填充
                    if (i+ii-sizepart)<0 or (i+ii-sizepart)>=image.shape[0]:
                        pass
                    elif (j+jj-sizepart)<0 or (j+jj-sizepart)>=image.shape[1]:
                        pass
                    else:
                        data.append(image[i+ii-sizepart][j+jj-sizepart])
            #取每个区域内的中位数
            image[i][j] = int(get_median(data))
            data=[]
    return image
def my_median_blur_RGB(image, size):
    (b ,r, g) = cv.split(image)
    blur_b = my_median_blur_gray(b, size)
    blur_r = my_median_blur_gray(r, size)
    blur_g = my_median_blur_gray(g, size)
    result = cv.merge((blur_b, blur_r, blur_g))
    return result

if __name__ == '__main__':
    image_test1 = cv.imread('apic27858.jpg')
    #调用自定义函数
    my_image_blur_median = my_median_blur_RGB(image_test1, 5)
    #调用库函数
    computer_image_blur_median = cv.medianBlur(image_test1, 5)
    fig = plt.figure()
    fig.add_subplot(131)
    plt.title('原图')
    plt.imshow(image_test1)
    fig.add_subplot(132)
    plt.title('自定义函数滤波')
    plt.imshow(my_image_blur_median)
    fig.add_subplot(133)
    plt.title('库函数滤波')
    plt.imshow(computer_image_blur_median)
    plt.show()