python 批量图片加噪声
时间: 2023-07-07 21:42:44 浏览: 72
您可以使用Python的Pillow库来批量添加噪声到图片中。以下是一个示例代码:
```python
from PIL import Image, ImageDraw
import random
import os
# 输入文件夹路径
input_folder = "input"
# 输出文件夹路径
output_folder = "output"
# 噪声强度
noise_intensity = 50
# 遍历文件夹中的所有图片
for filename in os.listdir(input_folder):
if filename.endswith(".jpg") or filename.endswith(".png"):
# 打开图片
img = Image.open(os.path.join(input_folder, filename))
# 获取图片宽度和高度
width, height = img.size
# 创建一个与图片大小相同的画布
draw = ImageDraw.Draw(img)
# 随机添加噪点到画布上
for x in range(width):
for y in range(height):
r, g, b = img.getpixel((x, y))
noise = random.randint(-noise_intensity, noise_intensity)
draw.point((x, y), (r+noise, g+noise, b+noise))
# 保存图片
img.save(os.path.join(output_folder, filename))
```
在以上示例代码中,我们使用了Pillow库的Image模块和ImageDraw模块。该代码会遍历文件夹中的所有图片,对每张图片进行噪声添加,并保存到输出文件夹中。您可以根据需要调整噪声强度大小。