中值滤波python代码
时间: 2023-10-12 15:16:54 浏览: 151
这是一个使用Python实现中值滤波的代码示例:
```python
import cv2
import numpy as np
def median_filter(img, kernel_size):
# 边缘填充
padding = (kernel_size - 1) // 2
padded_img = cv2.copyMakeBorder(img, padding, padding, padding, padding, cv2.BORDER_REPLICATE)
# 中值滤波
output_img = np.zeros(img.shape, np.uint8)
for i in range(padding, img.shape[0]+padding):
for j in range(padding, img.shape[1]+padding):
output_img[i-padding][j-padding] = np.median(padded_img[i-padding:i+padding+1, j-padding:j+padding+1])
return output_img
# 载入图像并显示
img = cv2.imread('image.jpg', cv2.IMREAD_GRAYSCALE)
cv2.imshow('Original Image', img)
# 中值滤波并显示
filtered_img = median_filter(img, 5)
cv2.imshow('Filtered Image', filtered_img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
该函数接受两个参数——输入图像和卷积核的大小,返回一个经过中值滤波处理后的图像。在函数内部,首先对输入图像进行边缘填充,然后对每个像素点及其周围的像素计算中值,并将结果保存到输出图像中。最后,返回输出图像。
注意,为了使代码更易读,这里使用了OpenCV库完成了一些边界填充和图像显示的操作,但这并不影响中值滤波本身的实现。
阅读全文