如何利用代码将第一个文件夹里的图片添加椒盐噪声后在图片名称.jpg前加入_salt后导出到第二个文件夹
时间: 2024-05-08 07:18:12 浏览: 16
你可以使用Python中的OpenCV库来实现这个功能。请参考以下代码:
```
import cv2
import os
# 定义函数来添加椒盐噪声
def add_salt_pepper_noise(image):
h, w, c = image.shape
# 定义噪声比例
noise_ratio = 0.05
# 添加椒盐噪声
noise_img = image.copy()
for i in range(h):
for j in range(w):
rdn = np.random.rand()
if rdn < noise_ratio/2:
noise_img[i][j] = 0
elif rdn < noise_ratio:
noise_img[i][j] = 255
return noise_img
# 定义第一个文件夹和第二个文件夹的路径
folder1_path = "/path/to/folder1"
folder2_path = "/path/to/folder2"
# 遍历第一个文件夹中的所有图片文件
for filename in os.listdir(folder1_path):
if filename.endswith(".jpg"):
# 读取图片
img = cv2.imread(os.path.join(folder1_path, filename))
# 添加椒盐噪声
noisy_img = add_salt_pepper_noise(img)
# 在文件名前加上"_salt"
new_filename = os.path.splitext(filename)[0] + "_salt.jpg"
# 保存图片到第二个文件夹
cv2.imwrite(os.path.join(folder2_path, new_filename), noisy_img)
```
这段代码会遍历第一个文件夹中的所有.jpg文件,对每个文件添加椒盐噪声,然后在文件名前加上"_salt",最后将处理后的图片保存到第二个文件夹中。