[发明专利]白平衡校正方法、装置和计算机设备

说明书

本申请涉及一种白平衡校正方法、装置和计算机设备，其中方法包括获取聚类字典，聚类字典通过对多个不同色温下的白色块图像进行聚类得到；计算待处理图像在聚类字典上的稀疏表示，并根据稀疏表示判断待处理图像的色温场景；色温场景为单色温场景或混合色温场景；根据待处理图像的色温场景，计算待处理图像中每一色温所占的权重；根据待处理图像中每一色温所占的权重，计算待处理图像中统计点的白平衡增益，并根据白平衡增益，对待处理图像进行白平衡校正。上述方法解决了相关技术中白平衡校正算法存在算法精度低、在混合色温下校正效果不精确的问题。

技术领域

本申请涉及图像处理领域，特别是涉及一种白平衡校正方法、装置和计算机设备。

背景技术

人眼具有独特的适应性，无论是在室外自然光下，还是在室内荧光灯或者混合色温光源下，人眼都会将白色物体识别为白色。由于摄像机对白色的响应与人眼不同，因此需要对摄像机采集到的图像进行白平衡校正，把采集到的图像中白色物体修正为白色。

现有的白平衡校正算法通常在某一色温下根据落入参考白区的白点计算出R/G/B三通道的增益，然后对画面中所有像素点分别乘以各个通道的增益，使画面中白色物体校正为真正的白色。这种算法在单一色温中效果表现良好，但在混合色温场景下无法对色温进行有效地判断，因此导致白平衡校正效果较差，难以满足要求。摄像机在室外实际使用时，会不可避免的遇到混合色温场景。现有的摄像机白平衡校正算法存在算法精度低、在混合色温下效果结果偏差大等缺点。

目前针对相关技术中的问题，尚未提出有效的解决方案。

发明内容

本申请实施例提供了一种白平衡校正方法、装置和计算机设备，以至少解决相关技术中白平衡校正算法存在算法精度低、在混合色温下校正效果不精确的问题。

第一方面，本申请实施例提供了一种白平衡校正方法，所述方法包括：

获取聚类字典，所述聚类字典通过对多个不同色温下的白色块图像进行聚类得到；

根据所述稀疏表示判断所述待处理图像的色温场景，并计算所述待处理图像中每一色温所占的权重；所述色温场景为单色温场景或混合色温场景；

根据所述待处理图像中每一色温所占的权重，计算所述待处理图像中统计点的白平衡增益，并根据所述白平衡增益，对所述待处理图像进行白平衡校正。

在其中一些实施例中，对多个不同色温下的白色块图像进行聚类得到聚类字典包括：

采集不同色温下的标准色卡图像中的白色块图像，确定基准白点，并根据所述基准白点绘制参考白区；

通过聚类算法计算出落入所述参考白区中统计点的聚类中心，并将多个所述聚类中心组合成所述聚类字典；所述白色块图像包括N个均等的第一图像块，每一个所述第一图像块为一个所述统计点。

在其中一些实施例中，所述计算待处理图像在所述聚类字典上的稀疏表示包括：

获取所述待处理图像中落入所述参考白区中的统计点，并将落入所述参考白区中的所述统计点拉伸成列向量；所述待处理图像划分为N个均等的第二图像块，每一个所述第二图像块为一个统计点；

计算每个所述第二图像块生成的列向量在所述聚类字典上的稀疏表示。

在其中一些实施例中，所述获取所述待处理图像中落入所述参考白区中的统计点包括：

获取每一所述第二图像块中像素点的三通道分量均值，根据所述三通道分量均值计算所述统计点的红色增益及蓝色增益；

根据所述统计点的红色增益及蓝色增益，确定落入所述参考白区中的统计点。