椒盐噪声与中值滤波的Python实现与检验

1.椒盐噪声(Salt Pepper Noise)

椒盐噪声也称为脉冲杂讯,是图像中经常见到的一种杂讯,它是一种随机出现的白点或者黑点,可能是亮的区域有黑色像素或是在暗的区域有白色像素(或是两者皆有) ——维基百科

1.1 关于椒盐噪声的几点注意

  • 噪点类型随机,即亮斑或暗斑(对应灰度图0、255)
  • 噪声概率为先验概率(如:噪声概率为0.1,数据点总数为100,而实际的噪点数并不一定为10)。关于先验概率与后验概率这篇博文比较推荐

1.2 实现算法思路

  1. 新建一个与输入图像等长的数组
  2. 数组中存放(1, \frac{100}{100\times\text{ noisy probability}})的随机整数
  3. 当上述数组中出现的值为1时,则认为对应的数据点为噪点
  4. 该噪点等概率的取亮斑或暗斑(0, 255)

1.3 Code

import numpy as np
import random

def salt_pepper(intput_signal, probability):
    '''
    椒盐噪声算法(适用于灰度图)
    :param intput_signal: 输入信号矩阵(2D)
    :param probability: 噪声概率(如:0.1为10%)
    :return: 加噪后的图像、实际噪点数、理论噪点数
    '''
    nisy = [0, 255] # 噪声(salt, papper)

    m, n= intput_signal.shape  # 获取输入图片尺寸(行和列)

    intput_signal_cp = intput_signal.copy() # 输入信号的副本
    intput_signal_cp = intput_signal_cp.reshape(-1) # reshape为一个行向量

    # 该噪声概率下,理论上noisy_data_probability_num个数据点中1一个噪点
    noisy_data_probability_num = int(100 / (probability * 100))

    # 噪点数组,当数组中的值为1时,则认为对应的数据点为噪点
    noisy_data = []
    for i in range(m*n):
        noisy_data.append(random.randint(1, noisy_data_probability_num))

    # 实际噪点数与理论噪点数
    actual_noisy_data_num = 0
    theory_noisy_data_num = int(m * n * probability)

    # 添加噪点
    for i in range(m*n):
        if noisy_data[i] == 1:
            actual_noisy_data_num = actual_noisy_data_num + 1
            intput_signal_cp[i] = nisy[random.randint(0, 1)]

    # 重塑至m*n的矩阵
    intput_signal_cp = intput_signal_cp.reshape(m, n)

    return  intput_signal_cp, actual_noisy_data_num, theory_noisy_data_num

2.中值滤波(Median Filtering)

中值滤波是一种非线性数字滤波器技术,经常用于去除图像或者其它信号中的噪声,是处理椒盐噪声的常用降噪算法——百度百科

2.1 窗口问题

中值滤波降噪的原理为使用噪点的某一个邻域内的中值代替该噪点,邻域范围越大则边缘保留效果越差,从视觉上来看,即更加模糊,但降噪效果明显。此外,邻域亦即窗口的形状也会影响降噪的效果。通常有3\times3、5\times5,十字形,圆形等。这里此文采用十字形。

2.2 Code

import numpy as np
import random

def median_filtering(input_signal):
    '''
    中值滤波(适用于灰度图)
    如:
            - + -
            + * +
            - + -
    * 为噪点,滤波方法为:取4个+的中位数(若某+不在输入信号范围内,则随机添加0或255)
    :param input_signal: 输入信号矩阵(2D)
    :return: 滤波后的信号
    '''
    salt_pepper = [0, 255]

    m, n = input_signal.shape  # 获取输入图片的尺寸(行和列)

    input_signal_cp = input_signal.copy()   # 输入信号的副本

    nosiy_data_around = []  # 存放噪点上下左右的数据点
    # 遍历滤波
    for i in range(m):
        for j in range(n):
            # 当灰度值为0或255时,则认为该数据点为椒盐噪点
            if input_signal_cp[i, j] == 255 or input_signal_cp[i, j] == 0:
                # 每次无效数据点(即不再范围内)为4,每有一个在范围内,即-1
                invalid_data_per = 4
                if i + 1 < n:
                    nosiy_data_around.append(input_signal_cp[i + 1, j])
                    invalid_data_per = invalid_data_per - 1
                if i - 1 >= 0:
                    nosiy_data_around.append(input_signal_cp[i - 1, j])
                    invalid_data_per = invalid_data_per - 1
                if j + 1 < m:
                    nosiy_data_around.append(input_signal_cp[i, j + 1])
                    invalid_data_per = invalid_data_per - 1
                if j - 1 >= 0:
                    nosiy_data_around.append(input_signal_cp[i, j - 1])
                    invalid_data_per = invalid_data_per - 1
                else:
                    if invalid_data_per > 0:
                        # 根据无效数据点的个数,随机添加0或255
                        for k in range(invalid_data_per):
                            nosiy_data_around.append(salt_pepper[random.randint(0, 1)])
                # 取中位数
                input_signal_cp[i, j] = np.median(nosiy_data_around)

                # 该噪点的周围数据数组清空,为下一个噪点周围数据存在做准备
                nosiy_data_around = []

    return input_signal_cp


3.实验对比

我这里采用Python第三方库skimage.data.camera()作为测试图片

椒盐噪率为0.1
椒盐噪率为0.0.3

4.Github

clike me!

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 218,755评论 6 507
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,305评论 3 395
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 165,138评论 0 355
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,791评论 1 295
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,794评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,631评论 1 305
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,362评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,264评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,724评论 1 315
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,900评论 3 336
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,040评论 1 350
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,742评论 5 346
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,364评论 3 330
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,944评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,060评论 1 270
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,247评论 3 371
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,979评论 2 355

推荐阅读更多精彩内容