批量对数增强的python代码
时间: 2023-11-12 07:44:43 浏览: 82
python进行数据增强
以下是一个简单的Python代码示例,展示了如何使用OpenCV库实现批量对数增强:
```python
import cv2
import numpy as np
import os
def batch_log_enhancement(input_folder, output_folder):
# 遍历输入文件夹中的所有图像文件
for filename in os.listdir(input_folder):
if filename.endswith(".jpg") or filename.endswith(".png"):
# 读取图像
image_path = os.path.join(input_folder, filename)
image = cv2.imread(image_path)
# 将图像从整数表示转换为浮点表示
image_float = image.astype(np.float32) / 255.0
# 对数变换
enhanced_image = np.log(1 + image_float)
# 将图像从浮点表示转换回整数表示
enhanced_image = (enhanced_image * 255).astype(np.uint8)
# 保存增强后的图像
output_path = os.path.join(output_folder, filename)
cv2.imwrite(output_path, enhanced_image)
# 设置输入和输出文件夹的路径
input_folder = "input_folder_path"
output_folder = "output_folder_path"
# 执行批量对数增强
batch_log_enhancement(input_folder, output_folder)
```
请注意,这只是一个简单的示例代码,您可能需要根据具体的需求进行修改和优化。此外,您还可以根据需要调整对数变换的参数和其他图像处理步骤来实现更灵活和准确的批量对数增强。
阅读全文