[发明专利]混合色温场景白平衡处理方法、装置、存储介质及终端

说明书

本发明提供了一种混合色温场景白平衡处理方法、装置、存储介质及终端，方法包括获取目标图像在UV直方图统计中的聚类结果，聚类结果包括聚类簇分布、聚类簇中心和聚类簇数目；根据聚类簇分布确定目标图像中的色温区域，根据聚类簇数量确定色温区域的数量，并根据色温区域的数量确定当前色温场景类型；根据色温区域对应的聚类簇中心计算每一个色温区域的颜色增益值；获取相邻的色温区域的分界线，并根据分界线按照预设带宽形成平滑处理过渡带；计算平滑处理过渡带中的像素点分别与边界围线的距离，并根据距离和相邻色温区域的颜色增益值计算平滑处理过渡带中的像素点的目标颜色增益值。本发明提高了混合色温场景下白平衡校正结果的准确性。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种混合色温场景白平衡处理方法、装置、存储介质及终端。

背景技术

白平衡是描述成像中红绿蓝混合色中，白色精确度的指标。目前传统的白平衡校正算法是基于全局校正处理的，也即一张图像的所有像素点采用统一的全局增益值。这种白平衡校正算法对于单一色温光源场景下的图像可以取得较好的效果。但是在实际情况下，往往会出现两种及以上的色温光源，对于这种混合色温场景，上述基于全局校正的白平衡算法无法同时兼顾不同色温情形，因此难以得到较好的白平衡效果。

针对混光场景，现有的白平衡校正方案主要可以分为两大类：第一类是采集不同色温场景下的中性色图像，绘制白区。通过判断目标图像的白区落点，进行色温估计和白平衡校正。但是这种方式需要在实验室场景下进行预设光源图像的采集，同时对白区落点的估计方法在实际情形中精度有限。第二类是对目标图像进行分割，获取目标子图像；通过对目标子图像进行色彩估计或光源方向判断，来实现对色温的估计和增益系数的计算。这种方法需预先对目标图像进行分割，分割精度难以保证；同时基于包含语义信息的分割结果进行不同色温区域划分在实际应用中有较强的限制性。

另一方面，上述针对混光场景的两类方法在rgb颜色空间进行处理时，易受光照强度变化的影响，从而影响不同色温区域的判断和划分；而且未考虑混合色温边界处的平滑处理，从而会影响最终白平衡结果的准确性。

因此，有必要提供一种新型的混合色温场景白平衡处理方法、装置、存储介质及终端以解决现有技术中存在的上述问题。

发明内容

本发明的目的在于提供一种混合色温场景白平衡处理方法、装置、存储介质及终端，提高了混合色温场景下白平衡校正结果的准确性。

第一方面，为实现上述目的，本发明的所述一种混合色温场景白平衡处理方法，所述方法包括：

获取目标图像在UV颜色空间直方图统计中的聚类结果，所述聚类结果包括聚类簇分布、聚类簇中心和聚类簇数量；

根据所述聚类簇分布确定所述目标图像中的色温区域，根据所述聚类簇数量确定所述色温区域的数量，并根据所述色温区域的数量确定当前色温场景的类型；

根据所述色温区域所对应的聚类簇中心，计算每一个所述色温区域的颜色增益值；

在确定当前色温场景为混合色温场景后，获取相邻的所述色温区域的分界线，并根据所述分界线按照预设带宽形成平滑处理过渡带，其中，所述平滑处理过渡带分别在相邻的所述色温区域形成边界围线；

计算所述平滑处理过渡带中的像素点分别与所述边界围线的距离，并根据所述距离和相邻所述色温区域的颜色增益值计算所述平滑处理过渡带中的像素点的目标颜色增益值。

本发明所述混合色温场景白平衡处理方法的有益效果在于：通过获取目标图像在UV颜色空间直方图统计的聚类结果以确定目标图像中不同的色温区域以及对应的色温，以对不同色温区域进行不同增益的白平衡处理，而且在UV直方图上进行聚类簇计算，能够有效减少亮度变化带来的误差，提高色温区域划分结果的准确性，同时利用平滑处理过渡带对相邻色温区域的交界处进行平滑处理，以实现平滑性的色彩增益过渡，提高混合色温场景下的白平衡校正结果的准确性。