经典非局部均值滤波(NLM)算法python实现(1)

经典非局部均值滤波(NLM)算法python实现(单通道图像版本)

概述:非局部均值(NL-means)是近年来提出的一项新型的去噪技术。该方法充分利用了图像中的冗余信息,在去噪的同时能最大程度地保持图像的细节特征。基本思想是:当前像素的估计值由图像中与它具有相似邻域结构的像素加权平均得到。

工作原理:请参照参考文献:https://blog.csdn.net/literacy_wang/article/details/106937535

以下是博主根据自己的一些个人理解和网上MATLAB代码翻译过来的代码, 适合单通道的图片场景:

# 传统的非局部均值滤波器
import cv2 
import numpy as np 

# 单通道
# f为相似窗口的半径, t为搜索窗口的半径, h为高斯函数平滑参数(一般取为相似窗口的大小)
def make_kernel(f):
    kernel = np.zeros((2*f+1, 2*f+1), np.float32)
    for d in range(1, f+1):
        kernel[f-d:f+d+1, f-d:f+d+1] += (1.0/((2*d+1)**2))

    return kernel/kernel.sum()

def NLmeans_filter(src, f, t, h):
    H, W = src.shape
    out = np.zeros((H, W), np.uint8)
    pad_length = f+t
    src_padding = np.pad(src, (pad_length, pad_length), mode='symmetric').astype(np.float32)
    kernel = make_kernel(f)
    h2 = h*h

    for i in range(0, H):
        for j in range(0, W):
            i1 = i + f + t
            j1 = j + f + t
            W1 = src_padding[i1-f:i1+f+1, j1-f:j1+f+1] # 领域窗口W1
            w_max = 0
            aver = 0
            weight_sum = 0
            # 搜索窗口
            for r in range(i1-t, i1+t+1):
                for c in range(j1-t, j1+t+1):
                    if (r==i1) and (c==j1):
                        continue
                    else:
                        W2 = src_padding[r-f:r+f+1, c-f:c+f+1] # 搜索区域内的相似窗口
                        Dist2 = (kernel*(W1-W2)*(W1-W2)).sum()
                        w = np.exp(-Dist2/h2)
                        if w > w_max:
                            w_max = w
                        weight_sum += w
                        aver += w*src_padding[r, c]
            aver += w_max*src_padding[i1, j1] # 自身领域取最大的权重
            weight_sum += w_max
            out[i, j] = aver/weight_sum

    return out

img = cv2.imread("lena256gs10.bmp", 0)
out = NLmeans_filter(img, 2, 5, 10)
cv2.imwrite("result.bmp", out)

输出展示:
加了标准差为10的高斯噪声的lena图像:

经典非局部均值滤波(NLM)算法python实现(1)_第1张图片
滤波结果:

经典非局部均值滤波(NLM)算法python实现(1)_第2张图片
参考博文:
https://blog.csdn.net/literacy_wang/article/details/106937535
https://blog.csdn.net/nineship/article/details/82109363
https://blog.csdn.net/u010839382/article/details/48229579

你可能感兴趣的:(python,opencv,机器学习,算法)