python opencv批量图像增强代码
时间: 2023-05-23 22:06:25 浏览: 206
图像增强OpenCV
4星 · 用户满意度95%
A:
以下是一个简单的 Python OpenCV 批量图像增强的示例代码:
```python
import os
import cv2
# 定义图像增强函数
def enhance_image(image):
# 在这里实现你的图像增强算法
# ...
return image
# 定义图像读取和保存函数
def process_image(image_file, output_dir):
# 读取图像
image = cv2.imread(image_file)
# 执行图像增强
image = enhance_image(image)
# 保存图像
filename = os.path.basename(image_file)
output_path = os.path.join(output_dir, filename)
cv2.imwrite(output_path, image)
# 批量处理图像
def batch_process_images(input_dir, output_dir):
# 列出所有图像文件
image_files = [os.path.join(input_dir, f) for f in os.listdir(input_dir) if f.endswith('.jpg')]
# 逐个处理图像
for image_file in image_files:
process_image(image_file, output_dir)
# 测试
input_dir = '/path/to/input/dir'
output_dir = '/path/to/output/dir'
batch_process_images(input_dir, output_dir)
```
在这个示例中,我们定义了一个 `enhance_image` 函数来实现图像增强算法。然后,我们定义了一个 `process_image` 函数来读取和保存单个图像。最后,我们使用 `batch_process_images` 函数来批量处理图像。该函数将列举输入目录中的所有 JPG 图像文件,并逐个处理它们。处理后的图像将保存在输出目录中。请注意,图像参数可以根据需要调整。
阅读全文