Python批量给图片添加椒盐噪声

 Python批量给图片添加椒盐噪声，其他噪声也可用同样的方式，修改部分代码即可。

上代码！！

mport os
import random
import numpy as np
import cv2


# 定义添加椒盐噪声的函数
def add_salt_pepper_noise(image, noise_ratio):
    h, w = image.shape[:2]
    noise_num = int(h * w * noise_ratio)
    for i in range(noise_num):
        row = np.random.randint(0, h)
        col = np.random.randint(0, w)
        if np.random.randint(0, 2) == 0:
            image[row, col] = 0
        else:
            image[row, col] = 255
    return image


# 设置输入和输出文件夹路径
input_folder ="C:/Users/ABC/Desktop/imagedata"
output_folder = "C:/Users/ABC/Desktop/123"

# 获取文件夹中所有JPG格式的图像文件
image_list = [os.path.join(input_folder, f) for f in os.listdir(input_folder) if f.endswith('.jpg')]

# 循环处理每个图像并保存到输出文件夹
for image_path in image_list:
    # 读取图像
    image = cv2.imread(image_path)

    # 添加椒盐噪声
    image_with_noise = add_salt_pepper_noise(image, noise_ratio=0.1) #noise_ratio的数值自己选定

    # 获取输出文件路径并保存图像
    filename = os.path.basename(image_path)
    output_path = os.path.join(output_folder, filename)
    cv2.imwrite(output_path, image_with_noise)

print("所有图像处理完成！")