【youcans 的 OpenCV 例程 300 篇】107. 退化图像的维纳滤波（Wiener filter）

6. 退化图像复原

图像复原是对图像退化的过程进行估计，并补偿退化过程造成的失真，以便获得未经退化的原始图像或原始图像的最优估值，从而改善图像质量的一种方法。

典型的图像复原方法是根据图像退化的先验知识建立退化模型，以退化模型为基础采用滤波等手段进行处理，使复原后的图像符合一定的准则，达到改善图像质量的目的。

因此，图像复原是沿着质量降低的逆过程来重现真实的原始图像，通过去模糊函数而去除图像模糊。

6.2 退化图像的维纳滤波（Wiener filter）

逆滤波对加性噪声特别敏感，使得恢复的图像几乎不可用（例程9.20中的退化图像未加入噪声项）。

最小均方误差滤波用来去除含有噪声的模糊图像，其目标是寻找未污染图像的一个估计，使均方误差最小：
$e2=E{(f−f^)2}e^2 = E\{ (f - \hat{f})^2 \}$
最小均方差滤波由 Wiener [1942] 首先提出，是最早提出的线性图像复原方法，因此称为维纳滤波。

信噪比 $SNR (f) = S (f) / N (f)$ ，是信息承载信号功率（未退化的原图像）水平与噪声功率水平的测度。低噪声图像的信噪比较高，高噪声图像的信噪比较低。

将复原后的图像视为“信号”，而复原图像与原图像的差视为“噪声”，则可以将空间域的信噪比定义为：
$SNR=∑x=0M−1∑y=0N−1f^(x,y)2/∑x=0M−1∑y=0N−1[f(x,y)−f^(x,y)]2SNR = \sum_{x=0}^{M-1} \sum_{y=0}^{N-1} \hat{f}(x,y)^2 / \sum_{x=0}^{M-1} \sum_{y=0}^{N-1} [f(x,y) -\hat{f}(x,y)]^2$
最小均方误差滤波的传递函数描述为：
$\frac{H^* (f) S(f)}{|H(f)|^2 S(f) + N(f)} = \frac{H^* (f)}{|H(f)|^2 S(f) + N(f)/S(f)}$
式中 $G (f) 、 H (f)$ 是 g 和 h 在频率域的傅里叶变换， $S (f) 、 N (f)$ 是信号 x(t)、噪声 n(t) 的功率谱。

用信噪比将上式进一步表示为：
$\frac{1}{H(f)} \frac{|H(f)|^2}{|H(f)|^2 + 1/SNR(f)}$
当处理白噪声时 $N(u,v)|^2$ 是一个常数，但未退化图像和噪声的功率谱通常未知或不能估计，则可用下式近似：
$F^(u,v)=1H(u,v)∣H(u,v)∣2∣H(u,v)∣2+KG(u,v)\hat{F}(u,v)= \frac{1}{H(u,v)} \frac{|H(u,v)|^2}{|H(u,v)|^2 + K} G(u,v)$
K 是加到 $H(u,v)|^2$ 的所有项上的一个规定常数。

例程 9.21: 维纳滤波（Wiener filter）

    # 9.21: 退化图像的维纳滤波 (Wiener filter)def getMotionDsf(shape, angle, dist):xCenter = (shape[0] - 1) / 2yCenter = (shape[1] - 1) / 2sinVal = np.sin(angle * np.pi / 180)cosVal = np.cos(angle * np.pi / 180)PSF = np.zeros(shape)  # 点扩散函数for i in range(dist):  # 将对应角度上motion_dis个点置成1xOffset = round(sinVal * i)yOffset = round(cosVal * i)PSF[int(xCenter - xOffset), int(yCenter + yOffset)] = 1return PSF / PSF.sum()  # 归一化def makeBlurred(image, PSF, eps):  # 对图片进行运动模糊fftImg = np.fft.fft2(image)  # 进行二维数组的傅里叶变换fftPSF = np.fft.fft2(PSF) + epsfftBlur = np.fft.ifft2(fftImg * fftPSF)fftBlur = np.abs(np.fft.fftshift(fftBlur))return fftBlurdef inverseFilter(image, PSF, eps):  # 逆滤波fftImg = np.fft.fft2(image)fftPSF = np.fft.fft2(PSF) + eps  # 噪声功率，这是已知的，考虑epsilonimgInvFilter = np.fft.ifft2(fftImg / fftPSF)  # 计算F(u,v)的傅里叶反变换imgInvFilter = np.abs(np.fft.fftshift(imgInvFilter))return imgInvFilterdef wienerFilter(input, PSF, eps, K=0.01):  # 维纳滤波，K=0.01fftImg = np.fft.fft2(input)fftPSF = np.fft.fft2(PSF) + epsfftWiener = np.conj(fftPSF) / (np.abs(fftPSF)**2 + K)imgWienerFilter = np.fft.ifft2(fftImg * fftWiener)imgWienerFilter = np.abs(np.fft.fftshift(imgWienerFilter))return imgWienerFilter# 读取原始图像img = cv2.imread("../images/Fig0526a.tif", 0)  # flags=0 读取为灰度图像hImg, wImg = img.shape[:2]# 不含噪声的运动模糊PSF = getMotionDsf((hImg, wImg), 45, 100)  # 运动模糊函数imgBlurred = np.abs(makeBlurred(img, PSF, 1e-6))  # 生成不含噪声的运动模糊图像imgInvFilter = inverseFilter(imgBlurred, PSF, 1e-6)  # 逆滤波imgWienerFilter = wienerFilter(imgBlurred, PSF, 1e-6)  # 维纳滤波# 带有噪声的运动模糊scale = 0.05  # 噪声方差noisy = imgBlurred.std() * np.random.normal(loc=0.0, scale=scale, size=imgBlurred.shape)  # 添加高斯噪声imgBlurNoisy = imgBlurred + noisy  # 带有噪声的运动模糊imgNoisyInv = inverseFilter(imgBlurNoisy, PSF, scale)  # 对添加噪声的模糊图像进行逆滤波imgNoisyWiener = wienerFilter(imgBlurNoisy, PSF, scale)  # 对添加噪声的模糊图像进行维纳滤波plt.figure(figsize=(9, 7))plt.subplot(231), plt.title("blurred image"), plt.axis('off'), plt.imshow(imgBlurred, 'gray')plt.subplot(232), plt.title("inverse filter"), plt.axis('off'), plt.imshow(imgInvFilter, 'gray')plt.subplot(233), plt.title("Wiener filter"), plt.axis('off'), plt.imshow(imgWienerFilter, 'gray')plt.subplot(234), plt.title("blurred image with noisy"), plt.axis('off'), plt.imshow(imgBlurNoisy, 'gray')plt.subplot(235), plt.title("inverse filter"), plt.axis('off'), plt.imshow(imgNoisyInv, 'gray')plt.subplot(236), plt.title("Wiener filter"), plt.axis('off'), plt.imshow(imgNoisyWiener, 'gray')plt.tight_layout()plt.show()