基于颜色直方图的非光源估计AWB方法及系统

技术领域

本发明涉及图像处理领域，涉及一种基于颜色直方图的非光源估计AWB方法及系统。

背景技术

人类视觉系统具有颜色恒常性特点，因此人类对物体观察不受光源影响。但是传感器在不同光线下，物体呈现的颜色不同，在晴朗天空下会偏蓝，在烛光下会偏红。为了消除光源对于图像传感器成像的影响，模拟人类视觉系统的颜色恒常性，要通过白平衡算法来对各种色温下的图像进行还原。

常见的基于光源估计的自动AWB，比较适用于单一光源的情况，但在多光源下，若仍使用单光源算法估计一个场景中的光源不能有效校正图像的白平衡；基于深度卷积神经网络的白平衡算法可以有效解决图像场景中的多光源以及光源分布不均的情况，但是这类算法需要大量的图片数据集，算法复杂且对硬件需求较高；但基于直方图处理的白平衡算法无需估计光源且算法的复杂性低。

发明内容

有鉴于此，本发明的目的在于提供一种基于颜色直方图的非光源估计AWB方法及系统，能够解决传统基于光源估计的自动白平衡算法场景适应性差、色彩校正不准确等问题，又能够改善多光源下图像自动白平衡的处理效果。

为实现上述目的，本发明采用如下技术方案：

一种基于颜色直方图的非光源估计AWB方法，包括以下步骤：

获取原始图像，并进行颜色直方图的RGB色度范围判断；

若RGB若色度范围大于预设则进行直方图匹配处理，并通过计算RGB通道重叠面积最大时的增益系数，然后使用线性处理来校正图像的白平衡；

若RGB若色度范围小于预设，则进行直方图相似度判断，若相似度小于预设值则进行拉伸处理，通过调整颜色直方图的色调范围，若相似度大于预设值进行平移处理。

进一步的，所述对图像进行颜色直方图的色度范围判断具体为：对RGB三通道的各灰度级及像素分布进行统计，将每个通道分成四个容器(0-63，64-127，128-191，192-255)，然后将每个通道的四个区间进行组合用来得到一个含有64元素的特征向量并用于特征匹配，根据实际情况将每个通道每个区间分配不同的权值。

进一步的，所述对RGB色度范围较宽图像对图像R通道和B通道的直方图分别匹配到G通道的直方图上，并利用颜色通道直方图的一致性来估计图像的白平衡，具体如下：

计算R、G通道的通道直方图，其中h

然后计算出R、G通道的累计直方图S

进一步的，所述拉伸处理，具体如下：分别定义U和L两个阈值，U为输入图像颜色直方图或百分位数的最大像素值，L为输入图像颜色直方图中的最小像素值或百分位数的最小像素值，U和L的计算公式表示为：

计算U和L的差值，若其差值大于阈值Th，则表明图像的灰度级分布区间较大。

进一步的，所述计算通道重叠面积最大时的增益系数，然后使用线性处理来校正图像的白平衡，具体如下：对图像集不同U、L值的图像进行计算，将U和L分别设置为高于其他99％像素的值和低于其他99％像素的值，并为其设定一个阈值Th，对于大于阈值Th的图像需要根据以下公式计算出输出像素的色度值：

式中，式中表示图片像素的输出值，表示图片像素的输入值，R是输出色度范围。

进一步的，所述对RGB颜色通道相似度的判断，具体如下：

将各通道直方图0至255灰度级分为4个区间，然后计算每个区间内的像素数量；若两个及以上区间的比值大于所设的最大阈值Th_max或者小于最小阈值Th_min，则表明该通道的直方图和G通道直方图相差过大，即图像中存在大量饱和像素会导致直方图匹配算法修正过度，需要对直方图匹配算法进行通道平移的改进处理。

进一步的，所述平移处理，具体如下：

过从0级到255级的求和得到重叠区域，计算RGB通道直方图的重叠面积公式表示为：

其中H表示r、g和b通道颜色直方图的重叠区域，该重叠区域源自三个颜色通道的所有灰度级的交点，重叠区域最大值的计算公式表示为：

H

如果重叠区域H达到最大值，则增益系数k

一种基于颜色直方图的非光源估计AWB系统,包括处理器、存储器以及存储在所述存储器上的计算机程序，所述处理器执行所述计算机程序时，具体执行如上所述的AWB方法中的步骤。

本发明与现有技术相比具有以下有益效果：

1、本发明能够解决传统基于光源估计的自动白平衡算法场景适应性差、色彩校正不准确等问题，又能够改善多光源下图像自动白平衡的处理效果。

2、本发明能够改善传统基于光源估计的AWB在处理多光源图像时效果不佳的问题，白平衡适用场景更加广泛。

附图说明

图1是本发明方法流程图；

图2是本发明一实施例中直方图算法处理后图像RGB通道的分布结果；

图3是本发明与传统光源估计方法的测试对比结果。

具体实施方式

下面结合附图及实施例对本发明做进一步说明。

请参照图1，本发明提供一种基于颜色直方图的非光源估计AWB方法，包括以下步骤：

获取原始图像，并进行颜色直方图的RGB色度范围判断；

若RGB若色度范围大于预设则进行直方图匹配处理，并通过计算RGB通道重叠面积最大时的增益系数，然后使用线性处理来校正图像的白平衡；

若RGB若色度范围小于预设，则进行直方图相似度判断，若相似度小于预设值则进行拉伸处理，通过调整颜色直方图的色调范围，若相似度大于预设值进行平移处理。有效避免了图片单色区域被破坏导致色调信息调整过大的情况，使基于直方图算法所处理的图像能够调整更广的色温范围。

参考图2，在本实施例中，对输入的图像进行处理获得RGB通道直方图具体包括对输入的图像进行通道的判断处理，然后获得颜色直方图；所述RGB色度范围较宽图像的直方图进行匹配处理具体为：

对RGB三通道的各灰度级及像素分布进行统计，将每个通道分成四个容器(0-63，64-127，128-191，192-255)，然后将每个通道的四个区间进行组合用来得到一个含有64元素的特征向量并用于特征匹配，所述对图像R通道和B通道的直方图分别匹配到G通道的直方图上，并利用颜色通道直方图的一致性来估计图像的白平衡。

在本实施例中，对RGB色度范围较宽图像对图像R通道和B通道的直方图分别匹配到G通道的直方图上，并利用颜色通道直方图的一致性来估计图像的白平衡，具体如下：

计算R、G通道的通道直方图，其中h

然后计算出R、G通道的累计直方图S

在本实施例中，针对RGB色度范围较窄图像的直方图进行拉伸处理，具体如下：分别定义U和L两个阈值，U为输入图像颜色直方图或百分位数的最大像素值，L为输入图像颜色直方图中的最小像素值或百分位数的最小像素值，U和L的计算公式表示为：

计算U和L的差值，若其差值大于阈值Th，则表明图像的灰度级分布区间较大。

在本实施例中，计算通道重叠面积最大时的增益系数，然后使用线性处理来校正图像的白平衡，具体如下：对图像集不同U、L值的图像进行计算，将U和L分别设置为高于其他99％像素的值和低于其他99％像素的值，并为其设定一个阈值Th，对于大于阈值Th的图像需要根据以下公式计算出输出像素的色度值：

式中，式中表示图片像素的输出值，表示图片像素的输入值，R是输出色度范围。

在本实施例中，对RGB颜色通道相似度的判断，具体如下：

将各通道直方图0至255灰度级分为4个区间，然后计算每个区间内的像素数量；若两个及以上区间的比值大于所设的最大阈值Th_max或者小于最小阈值Th_min，则表明该通道的直方图和G通道直方图相差过大，即图像中存在大量饱和像素会导致直方图匹配算法修正过度，需要对直方图匹配算法进行通道平移的改进处理。

在本实施例中，平移处理，具体如下：

过从0级到255级的求和得到重叠区域，计算RGB通道直方图的重叠面积公式表示为：

其中H表示r、g和b通道颜色直方图的重叠区域，该重叠区域源自三个颜色通道的所有灰度级的交点，重叠区域最大值的计算公式表示为：

H

如果重叠区域H达到最大值，则增益系数k

在本实施例中，还提供一种基于颜色直方图的非光源估计AWB系统,包括处理器、存储器以及存储在所述存储器上的计算机程序，所述处理器执行所述计算机程序时，具体执行如上所述的AWB方法中的步骤。

从图3中可以看出，与传统光源估计方法相比，此方法能够改善基于光源估计的AWB算法在处理多光源图像时效果不佳的问题，本发明在处理图像的白平衡时更准确，且能够处理的图像场景更丰富。

本领域内的技术人员应明白，本申请的实施例可提供为方法、系统、或计算机程序产品。因此，本申请可采用完全硬件实施例、完全软件实施例、或结合软件和硬件方面的实施例的形式。而且，本申请可采用在一个或多个其中包含有计算机可用程序代码的计算机可用存储介质(包括但不限于磁盘存储器、CD-ROM、光学存储器等)上实施的计算机程序产品的形式。

本申请是参照根据本申请实施例的方法、设备(系统)、和计算机程序产品的流程图和/或方框图来描述的。应理解可由计算机程序指令实现流程图和/或方框图中的每一流程和/或方框、以及流程图和/或方框图中的流程和/或方框的结合。可提供这些计算机程序指令到通用计算机、专用计算机、嵌入式处理机或其他可编程数据处理设备的处理器以产生一个机器，使得通过计算机或其他可编程数据处理设备的处理器执行的指令产生用于实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能的装置。

这些计算机程序指令也可存储在能引导计算机或其他可编程数据处理设备以特定方式工作的计算机可读存储器中，使得存储在该计算机可读存储器中的指令产生包括指令装置的制造品，该指令装置实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能。

这些计算机程序指令也可装载到计算机或其他可编程数据处理设备上，使得在计算机或其他可编程设备上执行一系列操作步骤以产生计算机实现的处理，从而在计算机或其他可编程设备上执行的指令提供用于实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能的步骤。

以上所述，仅是本发明的较佳实施例而已，并非是对本发明作其它形式的限制，任何熟悉本专业的技术人员可能利用上述揭示的技术内容加以变更或改型为等同变化的等效实施例。但是凡是未脱离本发明技术方案内容,依据本发明的技术实质对以上实施例所作的任何简单修改、等同变化与改型，仍属于本发明技术方案的保护范围。