一、实现方法

自适应中值滤波算法有两个处理层次:
**层次A:**若zmin<zmed<zmax,则转移到层次B;
否则,增Sxy尺 寸,
若Sxy<Smax,则重复层次A;否则,输出zmed。
**层次B:**若zmin<zxy<zmax,则输出zxy;否则,输出zmed。
既然是中值滤波,那处理椒盐这类冲激噪声肯定比较合适,自适应中值滤波能在有效去除冲激噪声的基础上,平滑其他非冲激噪声,减少失真,保留图像细节,而对于同等大小的中值滤波器,则会损失较多的细节。原理性内容书中有详细的介绍~
运行环境:Python+anaconda
直接上代码(因为是逐个像素操作,所以运行有点慢)


二、代码

# -*- coding:utf-8 -*-
"""
作者:YJH
日期:20211019"""
import matplotlib.pyplot as plt
import cv2 as cv
import numpy as np
# from functools import reduce               # 导入reduce,将一个函数作用在一个序列上,并且序列内容自动累计

# 显示汉字
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False


# 定义坐标数字字体及大小
def label_def():
    plt.xticks(fontproperties='Times New Roman', size=8)
    plt.yticks(fontproperties='Times New Roman', size=8)


# 读取图片
img_saltpep = cv.imread('ckt_saltpep_prob_pt25.tif', 0)               # 灰度图,噪声密度50%


def auto_median_filter(image, max_size):
    origen = 3                                                        # 初始窗口大小
    board = origen//2                                                 # 初始应扩充的边界
    # max_board = max_size//2                                         # 最大可扩充的边界
    copy = cv.copyMakeBorder(image, *[board]*4, borderType=cv.BORDER_DEFAULT)         # 扩充边界
    out_img = np.zeros(image.shape)
    for i in range(image.shape[0]):
        for j in range(image.shape[1]):
            def sub_func(src, size):                         # 两个层次的子函数
                kernel = src[i:i+size, j:j+size]
                # print(kernel)
                z_med = np.median(kernel)
                z_max = np.max(kernel)
                z_min = np.min(kernel)
                if z_min < z_med < z_max:                                 # 层次A
                    if z_min < image[i][j] < z_max:                       # 层次B
                        return image[i][j]
                    else:
                        return z_med
                else:
                    next_size = cv.copyMakeBorder(src, *[1]*4, borderType=cv.BORDER_DEFAULT)   # 增尺寸
                    size = size+2                                        # 奇数的核找中值才准确
                    if size <= max_size:
                        return sub_func(next_size, size)     # 重复层次A
                    else:
                        return z_med
            out_img[i][j] = sub_func(copy, origen)
    return out_img


if __name__ == '__main__':                                       # 运行当前函数

    img_auto_filter = auto_median_filter(img_saltpep, 7)
    img_median = cv.medianBlur(img_saltpep, 7)
    # img_re_median = cv.medianBlur(img_auto_filter, 3)                           # 两次3x3的中值滤波效果才能差不多
    plt.subplot(131), plt.imshow(img_saltpep, "gray"), plt.title('椒盐噪声密度50%', fontsize='small'), label_def()
    plt.subplot(132), plt.imshow(img_auto_filter, "gray"), plt.title('自适应中值滤波', fontsize='small'), label_def()
    plt.subplot(133), plt.imshow(img_median, "gray"), plt.title('中值滤波', fontsize='small'), label_def()
    # plt.subplot(144), plt.imshow(img_re_median, "gray"), plt.title('再中值滤波', fontsize='small'), label_def()

    plt.show()

运行结果

在这里插入图片描述
效果还是挺好的
目前正在学习中,欢迎大家批评指正~

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐