一种图像白平衡方法、电子设备和可读存储介质与流程

本技术涉及图像处理，特别是涉及一种图像白平衡方法、电子设备和可读存储介质。

背景技术：

1、白平衡用于补偿在特定光源下拍摄时出现的偏色现象。例如，同一白色物体，在高色温下呈现偏蓝，低色温下呈现偏黄，人视觉系统会自动矫正偏差。对于成像设备而言，则需要白平衡算法进行修正补偿。

2、同一画面中可能存在多个色温的光源，例如，太阳照射区域和阴影区，路灯和机动车灯等。现有的图像白平衡方式中，对于混合色温场景很难处理，无法使得画面中高色温区域的白色物体和低色温区域的白色物体均还原成白色。

技术实现思路

1、本技术至少提供一种图像白平衡方法、电子设备和可读存储介质。

2、本技术提供了一种图像白平衡方法，该方法包括：获取待校正图像中的参考像素点；基于参考像素点提取待校正图像对应的第一色温增益值和第二色温增益值，第一色温增益值对应的色温高于第二色温增益值对应的色温；根据第一色温增益值和第二色温增益值确定各参考像素点的混合增益值；基于混合增益值对待校正图像进行白平衡校正。

3、其中，基于参考像素点提取第一色温增益值和第二色温增益值包括：基于参考像素点的第一色温关联参数，从待校正图像中划分出第一色温参考区域和第二色温参考区域；基于第一色温参考区域包含的所有参考像素点的颜色分量，获取第一色温增益值，以及基于第二色温参考区域包含的所有参考像素点的颜色分量，获取第二色温增益值。

4、其中，基于参考像素点的第一色温关联参数，从待校正图像中划分出第一色温参考区域和第二色温参考区域包括：获取所有参考像素点的第一色温关联参数的集中趋势统计值，基于各参考像素点的第一色温关联参数与集中趋势统计值之间的大小关系，将各参考像素点所在的区域划分为第一色温参考区域和第二色温参考区域。

5、其中，颜色分量包括红色分量、蓝色分量和绿色分量，基于第一色温参考区域包含的所有参考像素点的颜色分量，获取第一色温增益值步骤，或者基于第二色温参考区域包含的所有参考像素点的颜色分量，获取第二色温增益值的步骤，包括：对参考区域包含的所有参考像素点的各类颜色分量分别求和得到红色分量和、绿色分量和、蓝色分量和；将绿色分量和、红色分量和之比作为增益值的第一分量，以及将绿色分量和、蓝色分量和之比作为增益值的第二分量；其中，参考区域为第一色温参考区域、增益值为第一色温增益值，或者，参考区域为第二色温参考区域、增益值为第二色温增益值。

6、其中，根据第一色温增益值和第二色温增益值确定各参考像素点的混合增益值包括：从参考像素点中筛选满足第一色温条件的第一参考像素点，和满足第二色温条件的第二参考像素点，其中，第一色温条件表示色温高于第一色温增益值对应的色温，第二色温条件表示色温低于第二色温增益值对应的色温；将第一色温增益值作为第一参考像素点的混合增益值，以及将第二色温增益值作为第二参考像素点的混合增益值。

7、其中，第一色温条件包括参考像素点的第二色温关联参数低于第一关联阈值，第二色温条件包括参考像素点的第二色温关联参数高于第二关联阈值，其中，第二色温关联参数与像素点的色温负相关，第一关联阈值为第一色温增益值等效的第二色温关联参数，第二关联阈值为第二色温增益值等效的第二色温关联参数。

8、其中，根据第一色温增益值和第二色温增益值确定各参考像素点的混合增益值包括：将至少部分参考像素点作为第三参考像素点；获取第三参考像素点对应的混合权重，其中，混合权重包括第一子权重和第二子权重，第一子权重和第二子权重中一者对应第一色温增益值，另一者对应第二色温增益值；利用混合权重对第一色温增益值和第二色温增益值进行加权求和，得到第三参考像素值的混合增益值。

9、其中，将至少部分参考像素点作为第三参考像素点，包括：将不满足第一色温条件以及第二色温条件的参考像素点作为第三参考像素点，其中，第一色温条件表示色温高于第一色温增益值对应的色温，第二色温条件表示色温低于第二色温增益值对应的色温。

10、其中，根据第一色温增益值和第二色温增益值确定各参考像素点的混合增益值之前，该方法还包括：基于待校正图像中包含的像素点的色温参数，对像素点进行聚类，得到若干类别的像素点，其中，色温参数包括白平衡增益值、色温值中的任一者；基于像素点的色温参数对各类别进行排序，将排序后的类别的序号作为类别中像素点的第三色温关联参数；获取第三参考像素点对应的混合权重包括：获取第三参考像素点的第三色温关联参数、第一色温增益值等效的第三色温关联参数、第二色温增益值等效的第三色温关联参数；基于第一差值与第二差值之间的比值得到第三参考像素点的第一子权重，基于第一子权重得到第二子权重；其中，第一差值为第三参考像素点的第三色温关联参数与第一色温增益值等效的第三色温关联参数之间的差值，或者第二色温增益值等效的第三色温关联参数与第三参考像素点的第三色温关联参数之间的差值，第二差值为第一色温增益值等效的第三色温关联参数、第二色温增益值等效的第三色温关联参数之间的差值。

11、其中，基于混合增益值对待校正图像进行白平衡校正之前，该方法还包括：从参考像素点中，选出与其他像素点对应的关联参考像素点，其中，其他像素点为待校正图像除参考像素点以外的像素点；基于关联参考像素点的颜色分量，得到其他像素点的混合增益值；基于混合增益值对待校正图像进行白平衡校正包括：基于待校正图像中各像素点的混合增益值对待校正图像进行白平衡校正。

12、其中，从参考像素点中，选出与其他像素点对应的关联参考像素点包括：分别在其他像素点的若干预设方向上选择参考像素点组，作为关联参考像素点。

13、其中，基于关联参考像素点的颜色分量，得到其他像素点的混合增益值包括：基于各参考像素点组与其他像素点之间的距离，分别确定各预设方向对应的参考权重，其中，参考权重与距离之间负相关；以及，基于参考像素点组包含的所有参考像素点的颜色分量，获取对应预设方向的白平衡增益值；利用参考权重对对应的预设方向的白平衡增益值进行加权求和，得到其他像素点的混合增益值。

14、其中，分别在其他像素点的若干预设方向上选择参考像素点组，作为关联参考像素包括：分别在各预设方向上查找与其他像素点之间的距离满足预设距离条件的参考像素点，作为各预设方向的定位像素点；对于各预设方向，利用预设方向的定位像素点，从待校正图像中确定目标区域，并将目标区域内的参考像素点作为对应预设方向的参考像素点组，其中，目标区域内包含定位像素点。

15、其中，基于混合增益值对待校正图像进行白平衡校正包括：将混合增益值与待校正图像的全局增益值进行融合，得到目标增益值；基于目标增益值对待校正图像进行白平衡校正。

16、其中，将混合增益值与待校正图像的全局增益值进行融合，得到目标增益值包括：基于第一色温增益值和第二色温增益值之间的差异，确定融合权重，其中，融合权重包括混合增益值对应的权重和全局增益值对应的权重，混合增益值对应的权重与差异正相关；利用融合权重对混合增益值和待校正图像的全局增益值进行加权求和，得到目标增益值；基于目标增益值对待校正图像进行白平衡校正包括：对目标增益值进行滤波处理，利用滤波后的目标增益值对待校正图像进行白平衡校正。

17、其中，获取待校正图像中的参考像素点包括：获取待校正图像中各像素点的属性信息，属性信息包括亮度、色调、饱和度、白平衡增益值；判断像素点的亮度分量是否满足预设亮度条件，色调和饱和度是否满足预设色度条件，白平衡增益值是否满足预设色温条件；响应于均满足，确定像素点为参考像素点；响应于有一者不满足，确定像素点为其他像素点，并将其他像素点的色温参数更新为预设数值，其中，色温参数包括白平衡增益值。

18、本技术还提供了一种电子设备，包括相互耦接的存储器和处理器，处理器用于执行存储器中存储的程序指令，以实现上述任一图像白平衡方法。

19、本技术还提供了一种计算机可读存储介质，其上存储有程序指令，程序指令被处理器执行时实现上述任一图像白平衡方法。

20、上述方案，第一色温增益值和第二色温增益值能够反映图像中的高低色温情况，在考虑到图像中的高低色温情况的基础上，基于第一色温增益值和第二色温增益值为各参考像素点确定对应的混合增益值，更能真实地反映色温，从而提升白平衡的效果。

21、应当理解的是，以上的一般描述和后文的细节描述仅是示例性和解释性的，而非限制本技术。