计算机视觉中的中值滤波：经典案例与Python代码解析

Hey小伙伴们！今天我们要聊的是计算机视觉中的一个重要技术——中值滤波。中值滤波是一种非线性滤波方法，主要用于去除图像中的椒盐噪声，同时保留图像的边缘和细节。通过中值滤波，我们可以显著改善图像的质量。让我们一起来看看如何使用Python实现中值滤波吧！🎉

📝 理论篇：中值滤波的基本原理

中值滤波是一种非线性滤波方法，通过将滤波器窗口内的像素值排序，取中间值作为新的像素值。这种方法特别适用于去除椒盐噪声，因为它不受极端值的影响。

1. 椒盐噪声（Salt and Pepper Noise）

椒盐噪声是一种常见的图像噪声，表现为图像中随机出现的黑色和白色像素点。中值滤波可以有效地去除这种噪声。

2. 中值滤波的工作原理

假设有一个 (3 ×3)的滤波器窗口，窗口内的像素值如下：
$\begin{matrix} 10 & 20 & 30\\ 40 & 50 & 60\\ 70 & 80 & 90\\ \end{matrix}$
中值滤波的步骤如下：

提取窗口内的像素值：[10, 20, 30, 40, 50, 60, 70, 80, 90]
排序：[10, 20, 30, 40, 50, 60, 70, 80, 90]
取中间值：[50]

新的像素值就是中间值 (50)。

📑 实战篇：使用Python实现中值滤波

接下来，我们通过一个具体的Python示例来实现中值滤波。我们将使用OpenCV库来处理图像，并使用NumPy进行矩阵运算。

1. 安装必要的库

首先，确保你已经安装了OpenCV和NumPy：

pip install opencv-python numpy

2. 读取和显示图像

我们先读取一张带有椒盐噪声的图像并显示它：

import cv2
import numpy as np# 读取图像
image = cv2.imread('path_to_your_image_with_noise.jpg')# 显示原始图像
cv2.imshow('Original Image with Noise', image)
cv2.waitKey(0)
cv2.destroyAllWindows()

3. 添加椒盐噪声

为了演示中值滤波的效果，我们先给图像添加一些椒盐噪声：

def add_salt_and_pepper_noise(image, prob):output = np.copy(image)num_salt = np.ceil(prob * image.size * 0.5).astype(int)num_pepper = np.ceil(prob * image.size * 0.5).astype(int)# Add Salt noisecoords = [np.random.randint(0, i - 1, num_salt) for i in image.shape[:2]]output[coords[0], coords[1]] = 255# Add Pepper noisecoords = [np.random.randint(0, i - 1, num_pepper) for i in image.shape[:2]]output[coords[0], coords[1]] = 0return output# 添加椒盐噪声
noisy_image = add_salt_and_pepper_noise(image, 0.05)# 显示带噪声的图像
cv2.imshow('Noisy Image', noisy_image)
cv2.waitKey(0)
cv2.destroyAllWindows()

4. 中值滤波

使用OpenCV的 medianBlur 函数进行中值滤波：

def median_filter(image, kernel_size=3):# 使用OpenCV的medianBlur函数进行中值滤波filtered_image = cv2.medianBlur(image, kernel_size)return filtered_image# 应用中值滤波
filtered_image = median_filter(noisy_image, kernel_size=5)# 显示滤波后的图像
cv2.imshow('Median Filtered Image', filtered_image)
cv2.waitKey(0)
cv2.destroyAllWindows()

5. 完整代码

将上述步骤整合在一起，完整的代码如下：

import cv2
import numpy as np# 读取图像
image = cv2.imread('path_to_your_image_with_noise.jpg')# 显示原始图像
cv2.imshow('Original Image with Noise', image)
cv2.waitKey(0)
cv2.destroyAllWindows()# 添加椒盐噪声
def add_salt_and_pepper_noise(image, prob):output = np.copy(image)num_salt = np.ceil(prob * image.size * 0.5).astype(int)num_pepper = np.ceil(prob * image.size * 0.5).astype(int)# Add Salt noisecoords = [np.random.randint(0, i - 1, num_salt) for i in image.shape[:2]]output[coords[0], coords[1]] = 255# Add Pepper noisecoords = [np.random.randint(0, i - 1, num_pepper) for i in image.shape[:2]]output[coords[0], coords[1]] = 0return output# 添加椒盐噪声
noisy_image = add_salt_and_pepper_noise(image, 0.05)# 显示带噪声的图像
cv2.imshow('Noisy Image', noisy_image)
cv2.waitKey(0)
cv2.destroyAllWindows()# 中值滤波
def median_filter(image, kernel_size=3):# 使用OpenCV的medianBlur函数进行中值滤波filtered_image = cv2.medianBlur(image, kernel_size)return filtered_image# 应用中值滤波
filtered_image = median_filter(noisy_image, kernel_size=5)# 显示滤波后的图像
cv2.imshow('Median Filtered Image', filtered_image)
cv2.waitKey(0)
cv2.destroyAllWindows()