图像噪声、去噪基本方法合集(Python实现)
    		       		warning:
    		            这篇文章距离上次修改已过437天,其中的内容可能已经有所变动。
    		        
        		                
                
import numpy as np
import cv2
 
def add_gaussian_noise(image, mean=0, sigma=25):
    """
    添加高斯噪声
    """
    noise = np.random.normal(mean, sigma, image.shape)
    noisy_image = image + noise
    return np.clip(noisy_image, 0, 255).astype(np.uint8)
 
def median_filter(image, kernel_size=3):
    """
    中值滤波去噪
    """
    return cv2.medianBlur(image, kernel_size)
 
def custom_kernel_filter(image, kernel):
    """
    使用自定义核进行滤波去噪
    """
    return cv2.filter2D(image, -1, kernel)
 
def main():
    # 读取图像
    image = cv2.imread('input.jpg', 0)
    
    # 添加高斯噪声
    noisy_image = add_gaussian_noise(image)
    
    # 中值滤波去噪
    median_filtered_image = median_filter(noisy_image)
    
    # 自定义核滤波去噪
    custom_kernel = np.array([[-1, -1, -1], [-1, 9, -1], [-1, -1, -1]])
    custom_filtered_image = custom_kernel_filter(noisy_image, custom_kernel)
    
    # 显示和保存结果
    cv2.imshow('Original', image)
    cv2.imshow('Noisy Image', noisy_image)
    cv2.imshow('Median Filtered Image', median_filtered_image)
    cv2.imshow('Custom Kernel Filtered Image', custom_filtered_image)
    
    cv2.imwrite('noisy_image.jpg', noisy_image)
    cv2.imwrite('median_filtered_image.jpg', median_filtered_image)
    cv2.imwrite('custom_filtered_image.jpg', custom_filtered_image)
    
    cv2.waitKey(0)
    cv2.destroyAllWindows()
 
if __name__ == '__main__':
    main()这段代码首先定义了添加高斯噪声和中值滤波去噪的函数,然后定义了一个主函数main来读取图像,应用噪声、进行去噪,并显示和保存结果。这个例子展示了如何使用Python和OpenCV库来处理图像噪声。
评论已关闭