python实现给文件夹下图片加入随机噪声并批量保存

import os
import cv2
import numpy as np
import random
proportion=0.0025#加入多少小噪声点，可以调大
def convert(input_dir, output_dir):
    for filename in os.listdir(input_dir):
        path = input_dir + "/" + filename#获取文件路径
        print("doing... ", path)
        noise_img = cv2.imread(path)#读取图片
        height, width = noise_img.shape[0], noise_img.shape[1]#获取高度宽度像素值
        num = int(height * width * proportion)#一个准备加入多少噪声小点
        for i in range(num):
            w = random.randint(0, width - 1)
            h = random.randint(0, height - 1)
            if random.randint(0, 1) == 0:
                noise_img[h, w] = 0
            else:
                noise_img[h, w] = 255
        cv2.imwrite(output_dir+'/'+filename,noise_img)
if __name__ == '__main__':
    input_dir = "G:/shuju"#输入数据文件夹
    output_dir = "G:/shuju/xin"#输出数据文件夹
    convert(input_dir, output_dir)

参考