OpenCV实现彩色图像的直方图均衡化

1、直方图均衡化

在OpenCV中,equalizeHist函数用于直方图均衡化(Histogram Equalization)。这是一种图像处理技术,旨在增强图像的对比度,特别是在图像的灰度值集中于某个范围时非常有用。通过调整图像的灰度分布,使得输出图像的灰度值更加均匀地分布在0到255的范围内,从而改善图像的视觉效果。
工作原理

直方图均衡化的基本思想是:

  1. 计算输入图像的直方图:统计每个灰度级出现的次数。
  2. 计算累积分布函数(CDF):对于每个灰度级,计算包括该灰度级在内的所有较低灰度级的像素总数。
  3. 使用CDF映射灰度级:将原始灰度级通过CDF映射到新的灰度级,确保输出图像的灰度值分布更加均匀。

2、OpenCV的直方图均衡化

python 复制代码
dst = cv2.equalizeHist(src)
  • src:输入图像,应为灰度图像。
  • dst:均衡化后的输出图像。

通过,参数的说明可知,这个API仅用于灰度图像,那么,如果想对彩色图像进行直方图均衡化,该怎么处理呢?

3、分通道直方图均衡化方案

顾名思义,把彩色图像的R、G、B三个通道分别进行直方图均衡化处理。python代码如下:

python 复制代码
def ChnEqhist(input):
    '''
    分通道进行直方图均衡化
    '''
    #B通道
    blue = cv.equalizeHist(input[:, :, 0])
    #G通道
    green = cv.equalizeHist(input[:, :, 1])
    #R通道
    red = cv.equalizeHist(input[:, :, 2])
    #通道合成
    result = np.stack((blue, green, red), axis=-1)
    return result

测试效果如下:

可看出,部分玩具的颜色还是整体泛白,存在失真的现象。那么,有没有改进的方案呢?

4、YUV颜色空间的直方图均衡化方案

算法步骤

  1. BGR颜色空间转YUV。
python 复制代码
 yuv = cv.cvtColor(input, cv.COLOR_BGR2YUV_I420)
  1. 单独对Y分量进行直方图均衡化。
python 复制代码
y_chn = yuv[0:height, 0:width]
#对Y分量进行直方图均衡
eq_y_chn = cv.equalizeHist(y_chn)
  1. 计算有分量直方图均衡后灰度值的变化系数 d e l t a delta delta。
python 复制代码
delta = (np.float32(eq_y_chn) / np.float32(y_chn))
  1. 使用 d e l t a delta delta系数对原始RGB空间的灰度进行相乘。
python 复制代码
dst = np.uint8(np.clip(np.multiply(input, delta[:, :, np.newaxis]), 0, 255))

测试图像如下:

5、效果对比和分析

1、在YUV空间进行直方图均衡化,能够减轻颜色失真的现象,避免了整体泛白。

2、也减小了算法的计算量,避免了3个通道的直方图均衡化操作。

相关推荐
深度学习lover22 分钟前
<数据集>yolo纸板缺陷识别数据集<目标检测>
python·深度学习·yolo·目标检测·计算机视觉·数据集
OAFD.22 分钟前
YLOLv4
人工智能·计算机视觉·目标跟踪
HVACoder44 分钟前
复习下线性代数,使用向量平移拼接两段线
c++·线性代数·算法
爱coding的橙子1 小时前
每日算法刷题Day77:10.22:leetcode 二叉树bfs18道题,用时3h
算法·leetcode·职场和发展
Swift社区1 小时前
LeetCode 404:左叶子之和(Sum of Left Leaves)
算法·leetcode·职场和发展
南枝异客1 小时前
查找算法-顺序查找
python·算法
QuantumLeap丶1 小时前
《数据结构:从0到1》-06-单链表&双链表
数据结构·算法
李牧九丶1 小时前
从零学算法59
算法
渡我白衣2 小时前
未来的 AI 操作系统(八)——灵知之门:当智能系统开始理解存在
人工智能·深度学习·opencv·机器学习·计算机视觉·语言模型·人机交互
一匹电信狗2 小时前
【C++】手搓AVL树
服务器·c++·算法·leetcode·小程序·stl·visual studio