[发明专利]光源颜色的确定方法、装置、存储介质及电子设备

说明书

本申请实施例公开了一种光源颜色的确定方法、装置、存储介质及电子设备。该方法包括：确定目标图像的目标特征向量；将所述目标特征向量输入分类器，得到所述特征向量与至少两个算法参数向量之间的匹配度；根据所述匹配度，从至少两个算法参数向量中确定目标算法参数向量；根据所述目标算法参数向量，确定所述目标图像的光源颜色。通过执行本技术方案，可以实现提高光源颜色的准确性，以及扩大光源颜色确定方法的适用范围的效果。

技术领域

本申请实施例涉及图像处理技术领域，尤其涉及一种光源颜色的确定方法、装置、存储介质及电子设备。

背景技术

在自然环境下，人的视觉系统具有抵制场景中光源颜色变化的能力。比如，对同一个场景，无论是在早晨偏黄的阳光照射下，还是旁晚时分偏红的阳光照射下，我们的视觉系统所感知到的场景的颜色始终保持恒定，这种能力也被称为视觉系统的颜色恒常性。

当前常见的光源颜色估计算法包括灰度世界法(Gray World)、完美反射法(max-RGB)等，这类方法原理简单、易于实现，但由于基于的假设太强，导致使用场景受限。例如对于灰度世界法，其假设是一幅彩色图像中的所有像素值的统计平均结果是灰色的，当遇到一些如大面积纯色的场景(如大面积黄色、蓝色等)时，这类算法的假设就明显不成立，导致对光源颜色的估计存在严重偏差；完美反射法的假设是画面中高亮区的颜色代表了光源颜色，这在一些光照均匀的场景(无明显高亮区)同样也会失效。因此，如何能够更加精确的确定图片的光源颜色，以及增加确定过程的适用场景，已经成为本领域技术人员亟待解决的技术难题。

发明内容

本申请实施例提供一种光源颜色的确定方法、装置、存储介质及电子设备，可以实现提高光源颜色的准确性，以及扩大光源颜色确定方法的适用范围的效果。

第一方面，本申请实施例提供了一种光源颜色的确定方法，该方法包括：

确定目标图像的目标特征向量；

将所述目标特征向量输入分类器，得到所述特征向量与至少两个算法参数向量之间的匹配度；

根据所述匹配度，从至少两个算法参数向量中确定目标算法参数向量；

根据所述目标算法参数向量，确定所述目标图像的光源颜色。

可选的，所述目标图像的目标特征向量包括目标图像的至少两阶图层特征向量；

相应的，确定目标图像的目标特征向量，包括：

将所述目标图像作为零阶图层；

对所述目标图像在空间域求导，得到高阶图层；

根据所述零阶图层和所述高阶图层，确定目标图像的目标特征向量。

可选的，将所述零阶图层和所述高阶图层由三维色度空间转换到二维色度空间；

在二维色度空间中按照预设规则划分子区域，将所述零阶图层在各个子区域中的像素点个数作为零阶图层特征向量；并将所述高阶图层在各个子区域中的像素点个数作为高阶图层特征向量；

将所述零阶图层特征向量与所述高阶图层特征向量组合形成完整特征向量；

或者，

将所述零阶图层和所述高阶图层由三维色度空间转换到二维色度空间；

确定所述零阶图层的第一特征向量，以及确定所述高阶图层的第一特征向量；其中，所述第一特征向量包括二维色度空间的平均色度值、最大色度值以及在二维色度空间中的分布标准差；