修改参数:
1.original_folder:需要为图像添加噪声的原始文件夹地址
2.new_folder:添加噪声后图像的保存文件夹地址
import os
import cv2
import random
if __name__ == '__main__':
# 原始文件夹路径
original_folder = r'C:\Users\12302\Desktop\text3\demo_one'
# 保存的新文件夹路径
new_folder = r'C:\Users\12302\Desktop\text3\demo_two'
for filename in os.listdir(original_folder):
img = cv2.imread(os.path.join(original_folder, filename),cv2.IMREAD_COLOR)
for num in range(100000):
x = random.randint(0, img.shape[0] - 1)
y = random.randint(0, img.shape[1] - 1)
if num % 2 == 0:
img[x, y] = 0
else:
img[x, y] = 255
cv2.imwrite(os.path.join(new_folder, filename), img, [int(cv2.IMWRITE_PNG_COMPRESSION), 5])
添加前:
添加后: