[发明专利]图像白平衡校正方法、装置、设备和存储介质

说明书

本发明涉及一种图像白平衡校正方法方法、装置，该方法包括：判断各采集周期得到的临时模板与预先确定正式模板是否匹配；如果有临时模板与正式模板不匹配，则以临时模板为目标模板，判断目标模板对应的采集周期之后的连续M个采集周期得到的M组临时模板中任一组临时模板与目标模板是否匹配；如果M组临时模板中任一组临时模板均与目标模板匹配，则更新正式模板，其中更新后的正式模板为目标模板；如果M组临时模板中至少一组临时模板与目标模板不匹配，则根据正式模板对临时模板进行白平衡校正。上述方法复杂度低，准确度较高，可以有效解决了夜晚交通摄像头画面中信号灯颜色偏色的问题。

技术领域

本发明涉及图像处理技术领域，特别是涉及图像白平衡校正方法、装置、计算机设备和存储介质。

背景技术

在交通领域，路口的各个设备都要和交通交通灯进行配合，例如交通摄像头监控种电警设备就需要在画面中有交通灯的画面来判断车辆违章，这就使交通灯的颜色的正确性在监控画面中有很大的影响。然而，夜晚有很多环境光，由于这些光的色温都不相同，摄像头感光器件会受周围光线的影响使监控画面中交通灯呈现的颜色有所偏差。

目前针对夜晚环境的白平衡算法有灰度世界和全反射，灰度世界法是假设一幅图像中所像素的色彩足够多，所以RGB(图像中每一个像素点都是由红绿蓝三种颜色块组成的)总量趋近于(R＝G＝B＝255)，这样就能用一整个图像RGB的平均值作为灰色的基准进行白平衡的调整，当然这种方法过于理想，在实际场景中很难实现；全反射法的核心就是假设一幅图像中亮度(R+G+B)最高的点就是白点，从而进行全局白平衡的调整，这种方法的缺陷也是显而易见的，夜晚环境中最亮的光源不一定就是白光。

综上所述，现有的白平衡算法技术在实际使用上显然存在不便与缺陷，所以有必要加以改进。

发明内容

基于此，有必要针对交通监控受环境光影响导致信号灯颜色偏色严重的问题，提供一种图像白平衡校正方法、装置、计算机设备和存储介质。

第一方面提供了一种图像白平衡校正方法，以预设时间间隔采集交通灯的图像，完成一个采集周期的图像采集，得到一组临时模板，其中一个所述采集周期内所述交通灯完成所有的交通指示颜色的一轮显示；所述图像的白平衡校正方法包括：

判断所述各采集周期得到的所述临时模板与预先确定正式模板是否匹配；

如果有所述临时模板与所述正式模板不匹配，则以所述临时模板为目标模板，判断所述目标模板对应的采集周期之后的连续M个采集周期得到的M组临时模板中任一组临时模板与所述目标模板是否匹配；

如果所述M组临时模板中任一组临时模板均与所述目标模板匹配，则更新所述正式模板，其中更新后的正式模板为所述目标模板；

如果所述M组临时模板中至少一组临时模板与所述目标模板不匹配，则根据所述正式模板对所述临时模板进行白平衡校正。

在一些实施中，所述方法还包括:预先确定所述正式模板；

所述预先确定所述正式模板包括：

完成一个初始采集周期的交通灯的图像的采集，得到一组初始临时模板；

判断所述初始临时模板对应的采集周期之后的连续N个采集周期得到的N组临时模板中任一组临时模板与所述初始临时模板是否匹配，

如果所述N组临时模板中任一组临时模板均与所述初始临时模板匹配，则将所述初始临时模板设置为正式模板；

如果所述N组临时模板中至少一组临时模板与所述初始临时模板不匹配，则将所述初始采集周期的下一个采集周期设置为新的初始采集周期。

在一些实施中，所述判断所述各采集周期得到的所述临时模板与预先确定正式模板是否匹配，包括：

判断所述临时模板和所述正式模板处于相对应的时间节点的图像是否匹配；