对HSI图像进行色调平衡代码
时间: 2023-10-03 08:07:00 浏览: 94
图像色彩校正代码
4星 · 用户满意度95%
以下是 Python 代码实现 HSI 图像的色调平衡:
```python
import cv2
import numpy as np
def histeq_HSI(img):
# 转换色彩空间为 HSI
hsi_img = cv2.cvtColor(img, cv2.COLOR_BGR2HSV_FULL)
h, s, i = cv2.split(hsi_img)
# 对亮度 i 进行直方图均衡化
i_equalized = cv2.equalizeHist(i)
# 合并 HSI 通道
hsi_equalized = cv2.merge((h, s, i_equalized))
# 转换色彩空间为 BGR
bgr_equalized = cv2.cvtColor(hsi_equalized, cv2.COLOR_HSV2BGR_FULL)
return bgr_equalized
```
以上代码中,`histeq_HSI` 函数接收一张 BGR 格式的图像,首先将其转换为 HSI 色彩空间,然后仅对亮度 i 进行直方图均衡化,最后将 HSI 通道合并并转换回 BGR 格式。
阅读全文