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

权利要求书

1.一种白平衡校正方法，其特征在于，所述方法包括：

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

计算待处理图像在所述聚类字典上的稀疏表示；

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

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

所述计算待处理图像在所述聚类字典上的稀疏表示包括：

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

将落入参考白区中的所述统计点拉伸成列向量；

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

所述根据所述稀疏表示判断所述待处理图像的色温场景包括：

计算所述第二图像块p在聚类字典D中字典原子d_i上的稀疏表示：p=a*d_i+ ；

若小于预设误差，则判定所述待处理图像中的第二图像块p为单色温场景；

若大于或等于预设误差，则判定所述待处理图像中的第二图像块p为混合色温场景，并根据稀疏表示的系数确定对应的多个所属色温；

所述表示用字典原子重构所述第二图像块p的重构误差。

2.根据权利要求1所述的方法，其特征在于，对不同色温下的白色块图像进行聚类得到聚类字典包括：

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

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

3.根据权利要求1所述的方法，其特征在于，所述获取所述待处理图像中落入所述参考白区中的统计点包括：

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

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

4.根据权利要求1所述的方法，其特征在于，所述聚类字典D={d₁, d₂,…, d_k}，其中k为色温的个数；所述计算每个所述第二图像块生成的列向量在所述聚类字典上的稀疏表示包括：

若m₁小于预设误差，则结束计算流程；

若m₁大于或等于预设误差，则在所述聚类字典中选择字典原子d_i，计算待处理图像块p在字典原子d_i和d_j上的稀疏表示：p=a*d_i+ b*d_j +m₂；依此类推，直至重构误差m₂小于或等于所述预设误差；

其中，a，b分别表示所述第二图像块p在字典原子d_i和d_j上的稀疏表示系数;

所述表示用字典原子重构所述第二图像块p的重构误差。

5.根据权利要求4所述的方法，其特征在于，根据所述稀疏表示判断所述待处理图像的色温场景包括：

获取所述第二图像块在稀疏程度为1的第一重构误差，以及所述待处理图像块p在稀疏程度为2的第二重构误差，并计算所述第一重构误差与所述第二重构误差的比值；

若所述比值小于预设阈值，则确定所述第二图像块p为单色温图像块，所属色温为字典原子d_i对应的色温；

若所述比值大于或等于所述预设阈值，则确定所述第二图像块p为混合温图像块，所属色温为字典原子d_i和d_j对应的色温。