[发明专利]一种自动白平衡优化方法、设备以及计算机可读存储介质

说明书

本申请公开了一种自动白平衡优化方法、设备以及计算机可读存储介质，该自动白平衡优化方法包括：获取图像的YUV颜色空间的第一编码数据；将YUV颜色空间的第一编码数据转换为预设颜色空间的第二编码数据；按照预设尺寸将图像划分为多个图像分块；利用所述第二编码数据对所述多个图像分块进行颜色识别，标记出预设颜色的图像分块；将标记出的图像分块的坐标数据以及图像进行自动白平衡处理。通过上述方法，本申请可以通过将图像转换到鲁棒性较好的其他颜色空间，然后对图像进行颜色识别，利用颜色识别结果提高自动白平衡处理的准确性。

技术领域

本申请涉及图像处理技术领域，特别是涉及一种自动白平衡优化方法、设备以及计算机可读存储介质。

背景技术

由于人眼具有独特的适应性，在不同的色温及光照下，人们观察到的世界颜色计划是一样的。但是，对于视频监控的摄像机，由于摄像机中的sensor传感器、镜头等因素不能够在某些环境变化下保持相对稳定的特性，所以摄像机无法在不同色温下均能够将图像的颜色作准确的判断，导致无法真实地监控客观事物。自动白平衡技术就是将摄像机在低色温下拍摄到的图像整体颜色偏黄、在高色温下拍摄到的图像整体颜色偏蓝的色偏进行校正。

目前主流的自动白平衡算法为基于先验知识的自动白平衡算法，基于先验知识的自动白平衡算法主要通过预先设置一个白区，将落入该区的统计点记作白点，那么没有落入该白区的统计点记作非白点。这种方式通过色域映射法则可以将颜色进行校正，该算法色偏校正精度高，适用范围广。

但是，监控摄像机在实际的应用场景，例如，室外监控区域存在大面积绿色时，基于先验知识的自动白平衡算法会将绿色统计点误作为白点，导致整体图像颜色偏色。

发明内容

本申请提供一种自动白平衡优化方法、设备以及计算机可读存储介质。

本申请提供的一个技术方案是：提供一种自动白平衡优化方法，所述自动白平衡优化方法包括：

获取图像的YUV颜色空间的第一编码数据；

将YUV颜色空间的第一编码数据转换为预设颜色空间的第二编码数据；

按照预设尺寸将所述图像划分为多个图像分块；

利用所述第二编码数据对所述多个图像分块进行颜色识别，标记出预设颜色的图像分块；

将标记出的图像分块的坐标数据以及图像进行自动白平衡处理。

在一些可能的实施例中，所述利用所述第二编码数据对所述多个图像分块进行颜色识别，标记出预设颜色的图像分块的步骤，包括：

获取所述第二编码数据的至少一个通道数据；

将所述至少一个通道数据输入预先训练的分类器，其中，所述分类器事先对所述预设颜色进行训练；

根据所述分类器的输出结果获取符合所述预设颜色的图像分块的坐标数据。

在一些可能的实施例中，所述自动白平衡优化方法，还包括：

提取所述图像中符合所述预设颜色的特征数据；

利用所述特征数据训练所述分类器；

其中，所述分类器为Weighted-KNN分类器。

在一些可能的实施例中，所述预设颜色空间为HSV颜色空间；

所述将所述至少一个通道数据输入预先训练的分类器的步骤，包括：

提取所述HSV颜色空间的第二编码数据的H通道数据；

计算每个图像分块的H通道数据的众数；

将所述每个图像分块对应的众数输入所述预先训练的分类器；