python将同一文件夹下的图片批量进行增强 代码
时间: 2023-12-31 21:04:12 浏览: 86
Python批量重命名同一文件夹下文件的方法
以下是使用Python Pillow库进行图像增强的示例代码:
```python
from PIL import Image, ImageEnhance
import os
#定义增强函数
def enhance_image(image_path, output_path):
#打开图像
image = Image.open(image_path)
#增强对比度
contrast_enhancer = ImageEnhance.Contrast(image)
contrast = 1.5
image = contrast_enhancer.enhance(contrast)
#增强颜色
color_enhancer = ImageEnhance.Color(image)
color = 1.5
image = color_enhancer.enhance(color)
#增强亮度
brightness_enhancer = ImageEnhance.Brightness(image)
brightness = 1.5
image = brightness_enhancer.enhance(brightness)
#保存增强后的图像
image.save(output_path)
#获取文件夹中所有图片的路径
image_folder_path = "path/to/folder"
image_paths = [os.path.join(image_folder_path, file_name) for file_name in os.listdir(image_folder_path) if file_name.endswith(".jpg")]
#对每个图像进行增强并保存
for image_path in image_paths:
file_name, extension = os.path.splitext(image_path)
output_path = file_name + "_enhanced" + extension
enhance_image(image_path, output_path)
```
该代码会将指定文件夹中所有以".jpg"结尾的图片进行增强,并在同一文件夹下以"_enhanced"为后缀保存增强后的图片。您可以根据需要调整增强参数。
阅读全文