[发明专利]一种图像白平衡方法及系统、存储介质及终端设备

说明书

本发明提供一种图像白平衡方法及系统、存储介质及终端设备，包括以下步骤：基于暗通道先验算法获取目标图像中的白色区域；建立从所述目标图像到还原图像的映射模型，所述映射模型包括待确定的映射参数；基于灰度世界算法，利用所述目标图像和所述白色区域获取所述映射模型的映射参数，以得到训练好的映射模型；基于所述目标图像和所述训练好的映射模型获取还原图像。本发明的图像白平衡方法及系统、存储介质及终端设备解决了现有技术中图像白平衡算法失效率高，基于的假设单一，硬件复杂度高等缺陷。

技术领域

本发明涉及图像处理的技术领域，特别是涉及一种图像白平衡方法及系统、存储介质及 终端设备。

背景技术

随着人工智能的白热化发展，机器已经为人类带来了巨大的便利。在人工智能领域中， 机器视觉将成为人工智能下一个主要发展方向。

人眼看到的物体颜色就是物体本身的真实颜色，且人眼所看到物体的颜色不容易受到周 围光线的影响，这种现象叫做色彩恒常性。图像传感器是计算机和其他电子设备获取图像信 息的电子元件，是机器视觉的信息入口，能够将光学图像转变为电信号，从而帮助其他元件 实现对图像的存储和机器视觉算法后期的处理。然而在获取图像时，图像传感器会受到环境 光的影响，会将它“看到”的所有可见光都存储下来。例如，在偏黄的光线下拍摄的图像， 如果不经过色彩还原处理，图像传感器会存储偏黄的图像，而不是物体本身的颜色。若直接 在显示屏上显示，图像就会偏黄，而正确的色彩对机器视觉来说也极为重要。因此，需要在 图像传感器存储图像后或在显示器显示图像前进行色彩还原以减少色偏，恢复图像本身的颜 色。

现有技术中提出了很多色彩还原的相关算法，其中最经典的色彩还原算法博客以下两种：

(1)灰度世界算法

灰度世界算法假设对于一副色彩丰富的图像而言，整个场景的平均反射是消色差的，即 各个颜色通道的平均值是相等的，从而在此基础上计算各通道增益。但是，当图像颜色较为 单一时，该算法失效，无法还原物体真实色彩。

(2)完美反射算法

完美反射算法假设图像上最亮点就是白点，并以此白点为参考计算各通道增益。但是， 当图像不存在白色区域时，该算法失效。

发明内容

鉴于以上所述现有技术的缺点，本发明的目的在于提供一种图像白平衡方法及系统、存 储介质及终端设备，用于解决现有技术中图像白平衡算法失效率高，基于的假设单一，硬件 复杂度高等缺陷。

为实现上述目的及其他相关目的，本发明提供一种基于暗通道先验与灰度世界算法的图 像白平衡方法，包括以下步骤：基于暗通道先验算法获取目标图像中的白色区域；建立从所 述目标图像到还原图像的映射模型，所述映射模型包括待确定的映射参数；基于灰度世界算 法，利用所述目标图像和所述白色区域获取所述映射模型的映射参数，以得到训练好的映射 模型；基于所述目标图像和所述训练好的映射模型获取还原图像。

于本发明一实施例中，基于暗通道先验算法获取目标图像中的白色区域包括以下步骤：

获取目标图像中每个像素点x在r、g、b通道上的像素值I^c(x)，c∈{r、g、b}；

根据获取暗通道的目标图像，I^dark(x)表示暗通道的目标图像中像 素点x的像素值；

将所述暗通道的目标图像中最亮的点的像素值作为大气光强的值A；

获取每个像素点x的透射率

计算所述目标图像的平均透射率t_average；

根据I(x)＝J(x)t(x)+A(1-t(x))获取J(x)，其中I(x)表示所述目标图像中的像素点x的像 素值；J(x)表示还原图像中的像素点x的亮度值；