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

权利要求书

1.一种基于暗通道先验与灰度世界算法的图像白平衡方法，其特征在于，包括以下步骤：

基于暗通道先验算法获取目标图像中的白色区域；

建立从所述目标图像到还原图像的映射模型，所述映射模型包括待确定的映射参数；

基于灰度世界算法，利用所述目标图像和所述白色区域获取所述映射模型的映射参数，以得到训练好的映射模型；

基于所述目标图像和所述训练好的映射模型获取还原图像；

基于暗通道先验算法获取目标图像中的白色区域包括以下步骤：

获取目标图像中每个像素点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的亮度值；

计算还原图像中的像素点的平均亮度值J_average；

遍历所有像素点，将满足t(x)t_average且J(x)J_average的像素点构成白色区域；所述映射模型为

其中，I_r(x)、I_g(x)、I_b(x)分别表示目标图像中像素点x在r、g、b通道上的像素值；I_r'(x)、I_g'(x)、I_b'(x)分别表示还原图像中像素点x在r、g、b通道上的像素值；u_r、u_g、u_b、v_r、v_g、v_b为映射参数；

基于灰度世界算法，利用所述目标图像和所述白色区域获取所述映射模型的映射参数，以得到训练好的映射模型包括以下步骤：

a)计算目标图像的平均灰度值gray；

b)根据灰度世界算法得到其中MN表示像素点的总个数；

c)计算白色区域的平均灰度值gray_max；

d)根据灰度世界算法得到

e)利用步骤b)和步骤d)中的公式计算所述映射模型的映射参数u_r、u_g、u_b、v_r、v_g、v_b，以得到训练好的映射模型。