Python中的斑点(Lee Filter)

时间:2016-09-30 07:27:02

标签: python python-3.x image-processing imagefilter sar

我正在尝试在卫星SAR图像中去除斑点噪声。我没有得到任何在SAR图像中去除斑点噪声的包。我已经尝试过pyradar,但它适用于python 2.7,我在Windows上运行带有python 3.5的Anaconda。此外,Rsgislib可用,但它在Linux上。 Joseph meiring还在github上提供了一个Lee过滤器代码,但它无法正常工作。 : https://github.com/reptillicus/LeeFilter

请注意,任何人都可以共享散斑滤镜的python脚本或如何在python中进行散斑滤镜设计。

2 个答案:

答案 0 :(得分:6)

这是一个有趣的小问题。而不是试图为它找到一个库,为什么不从定义中写出来呢?

from scipy.ndimage.filters import uniform_filter
from scipy.ndimage.measurements import variance

def lee_filter(img, size):
    img_mean = uniform_filter(img, (size, size))
    img_sqr_mean = uniform_filter(img**2, (size, size))
    img_variance = img_sqr_mean - img_mean**2

    overall_variance = variance(img)

    img_weights = img_variance / (img_variance + overall_variance)
    img_output = img_mean + img_weights * (img - img_mean)
    return img_output

如果您不希望窗口是x大小的正方形,只需将uniform_filter替换为其他内容(使用磁盘卷积,高斯滤波器等)。任何类型的(加权)平均滤波器都可以,只要它与计算img_meanimg_square_mean相同。

Lee过滤器似乎过时了作为过滤器。它在边缘处的表现不佳,因为对于任何具有边缘的窗口,方差将远远高于整体图像方差,因此权重(未过滤图像相对于过滤图像的权重)将进行接近1.

一个例子:

from pylab import *
import numpy as np
img = np.random.normal(0.5, 0.1, (100,100))
img[:,:50] += 0.25
imshow(img, vmin=0, vmax=1, cmap='gray')
imshow(lee_filter(img, 20), vmin=0, vmax=1, cmap='gray')

Noisy image with edge Lee filtered

正如你所看到的,一般来说降噪效果非常好,但沿边缘的噪音要弱得多。

我不熟悉SAR所以我不知道Lee滤镜是否具有某些特性,使其特别适合SAR中的散斑,但您可能希望研究现代边缘感知降噪器,如导向过滤器或双边过滤

答案 1 :(得分:0)

通常,很难在2D图中肉眼看到噪声滤波器的效果。 让我通过示例进行演示。假设我们有这张嘈杂的图片:

noise example

现在让我将此图像转换为3d网格图。然后它将看起来像这样。噪点变得非常清晰,但图片的左侧和右侧之间的深度也不同。

mesh plot

findpeaks包含许多过滤器,这些过滤器可从各种(旧python 2)库中使用,并被重写为python3。重写过滤器非常容易,如下所示。此示例对于SAR图像似乎不太具有代表性。在此示例中,均值或中值滤波器似乎表现很好。在局部高度很重要的斑点噪声图像中,这种均值/中值滤波器可以轻松去除峰值。

安装方式:

pip install findpeaks

运行方式:

from findpeaks import findpeaks

# Read image
img = cv2.imread('noise.png')

filters = [None, 'lee','lee_enhanced','kuan', 'fastnl','bilateral','frost','median','mean']

for getfilter in filters:
    fp = findpeaks(method='topology', scale=False, denoise=getfilter, togray=True, imsize=False, window=15)
    fp.fit(img)
    fp.plot_mesh(wireframe=False, title=str(getfilter), view=(30,30))

Comparison methods

如果您直接想使用降噪滤波器,可以按照以下步骤进行操作:

import findpeaks
import matplotlib.pyplot as plt

# Read image
img = cv2.imread('noise.png')

# filters parameters
# window size
winsize = 15
# damping factor for frost
k_value1 = 2.0
# damping factor for lee enhanced
k_value2 = 1.0
# coefficient of variation of noise
cu_value = 0.25
# coefficient of variation for lee enhanced of noise
cu_lee_enhanced = 0.523
# max coefficient of variation for lee enhanced
cmax_value = 1.73

# Some pre-processing
# Make grey image
img = findpeaks.stats.togray(img)
# Scale between [0-255]
img = findpeaks.stats.scale(img)

# Denoising
# fastnl
img_fastnl = findpeaks.stats.denoise(img, method='fastnl', window=winsize)
# bilateral
img_bilateral = findpeaks.stats.denoise(img, method='bilateral', window=winsize)
# frost filter
image_frost = findpeaks.frost_filter(img, damping_factor=k_value1, win_size=winsize)
# kuan filter
image_kuan = findpeaks.kuan_filter(img, win_size=winsize, cu=cu_value)
# lee filter
image_lee = findpeaks.lee_filter(img, win_size=winsize, cu=cu_value)
# lee enhanced filter
image_lee_enhanced = findpeaks.lee_enhanced_filter(img, win_size=winsize, k=k_value2, cu=cu_lee_enhanced, cmax=cmax_value)
# mean filter
image_mean = findpeaks.mean_filter(img, win_size=winsize)
# median filter
image_median = findpeaks.median_filter(img, win_size=winsize)


plt.figure(); plt.imshow(img_fastnl, cmap='gray'); plt.title('Fastnl')
plt.figure(); plt.imshow(img_bilateral, cmap='gray'); plt.title('Bilateral')
plt.figure(); plt.imshow(image_frost, cmap='gray'); plt.title('Frost')
plt.figure(); plt.imshow(image_kuan, cmap='gray'); plt.title('Kuan')
plt.figure(); plt.imshow(image_lee, cmap='gray'); plt.title('Lee')
plt.figure(); plt.imshow(image_lee_enhanced, cmap='gray'); plt.title('Lee Enhanced')
plt.figure(); plt.imshow(image_mean, cmap='gray'); plt.title('Mean')
plt.figure(); plt.imshow(image_median, cmap='gray'); plt.title('Median')

如果您想使用该库,可以在here上找到更多示例。