添加椒盐噪声代码python

当我们在图像处理中需要添加椒盐噪声时，可以使用Python编写代码来实现。下面是一个简单的示例代码：

import numpy as np
import cv2

def add_salt_pepper_noise(image, noise_ratio):
    height, width = image.shape[:2]
    noise_num = int(height * width * noise_ratio)
    
    for i in range(noise_num):
        x = np.random.randint(0, width)
        y = np.random.randint(0, height)
        if np.random.random() < 0.5:
            image[y, x] = 0  # 添加椒噪声
        else:
            image[y, x] = 255  # 添加盐噪声
    
    return image

# 读取图像
image = cv2.imread('image.jpg', cv2.IMREAD_GRAYSCALE)

# 添加椒盐噪声
noisy_image = add_salt_pepper_noise(image, 0.05)

# 显示原始图像和添加噪声后的图像
cv2.imshow('Original Image', image)
cv2.imshow('Noisy Image', noisy_image)
cv2.waitKey(0)
cv2.destroyAllWindows()

上述代码中，我们首先定义了一个`add_salt_pepper_noise`函数，该函数接受一个灰度图像和噪声比例作为参数。然后，根据噪声比例计算需要添加的噪声点数，并使用随机数生成器在图像中随机选择位置添加椒盐噪声。最后，我们使用OpenCV库显示原始图像和添加噪声后的图像。