书写一段代码使得可以批量处理几百张图片使得它光照增强
时间: 2023-09-02 21:14:47 浏览: 83
批量处理图片
这里给您提供一个Python脚本,使用OpenCV库来实现对图像的光照增强,您可以将待处理的图片放在指定的文件夹中,执行脚本后,会对所有图片进行光照增强处理,并保存到指定的文件夹中。
```python
import cv2
import os
# 定义待处理图片所在文件夹的路径
input_dir = './input'
# 定义处理后的图片保存的文件夹路径
output_dir = './output'
# 循环处理每张图片
for filename in os.listdir(input_dir):
# 读取待处理图片
img_path = os.path.join(input_dir, filename)
img = cv2.imread(img_path)
# 对图片进行光照增强处理
clahe = cv2.createCLAHE(clipLimit=2.0, tileGridSize=(8,8))
img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
img = clahe.apply(img)
# 保存处理后的图片
output_path = os.path.join(output_dir, filename)
cv2.imwrite(output_path, img)
```
上述脚本中,我们使用了OpenCV中的CLAHE算法来实现光照增强,具体实现步骤为:
1. 循环读取待处理图片的路径;
2. 使用`cv2.imread()`函数读取待处理图片;
3. 使用`cv2.createCLAHE()`函数创建CLAHE对象;
4. 将图片从BGR色彩空间转换为灰度色彩空间;
5. 对灰度图像使用CLAHE算法进行光照增强;
6. 使用`cv2.imwrite()`函数将处理后的图片保存到指定路径。
您可以根据需要修改待处理图片和处理后图片的文件夹路径,以及调整CLAHE算法的参数来获得更好的光照增强效果。
阅读全文