python实现给指定文件夹的图片加椒盐噪音
时间: 2023-08-13 20:04:02 浏览: 93
可以使用Python的PIL库来给指定文件夹中的图片加上椒盐噪音。下面是一个简单的实现示例:
```python
from PIL import Image
import os
import random
def add_salt_pepper_noise(image_path, output_path, probability=0.05):
image = Image.open(image_path)
width, height = image.size
pixels = image.load()
for y in range(height):
for x in range(width):
if random.random() < probability:
if random.random() < 0.5:
pixels[x, y] = (0, 0, 0) # 添加黑色噪音
else:
pixels[x, y] = (255, 255, 255) # 添加白色噪音
image.save(output_path)
def add_salt_pepper_noise_to_folder(input_folder, output_folder, probability=0.05):
if not os.path.exists(output_folder):
os.makedirs(output_folder)
for file_name in os.listdir(input_folder):
if file_name.endswith(".jpg") or file_name.endswith(".png"):
image_path = os.path.join(input_folder, file_name)
output_path = os.path.join(output_folder, file_name)
add_salt_pepper_noise(image_path, output_path, probability)
# 使用示例:
input_folder = "input_folder"
output_folder = "output_folder"
probability = 0.05
add_salt_pepper_noise_to_folder(input_folder, output_folder, probability)
```
上述代码中,`add_salt_pepper_noise`函数用于给单张图片添加椒盐噪音,`add_salt_pepper_noise_to_folder`函数用于遍历指定文件夹中的图片,并将加噪后的图片保存到指定的输出文件夹中。你可以将`input_folder`和`output_folder`替换为你实际的文件夹路径,`probability`参数可以控制噪音的密度,数值越大噪音越多。
阅读全文
相关推荐















