-- coding: utf-8 --

import numpy as np import cv2

读取图像

img = cv2.imread('lena.png', cv2.IMREAD_GRAYSCALE)

添加椒盐噪声

img_noise = cv2.imread('lena_salt_pepper.png', cv2.IMREAD_GRAYSCALE)

均值滤波

kernel_sizes = [3, 5, 7] for kernel_size in kernel_sizes: kernel = np.ones((kernel_size, kernel_size), np.float32) / (kernel_size * kernel_size) img_mean = cv2.filter2D(img_noise, -1, kernel) cv2.imshow('Mean Filter ({}x{})'.format(kernel_size, kernel_size), img_mean)

中值滤波

kernel_sizes = [3, 5, 7] for kernel_size in kernel_sizes: img_median = cv2.medianBlur(img_noise, kernel_size) cv2.imshow('Median Filter ({}x{})'.format(kernel_size, kernel_size), img_median)

cv2.waitKey(0) cv2.destroyAllWindows(