一种AI自动白平衡的算法和电子设备制造技术

本申请提供一种AI自动白平衡的算法和电子设备。其中，所述算法包括：将图像分别通过第一FFCC模型、第二FFCC模型、第三FFCC模型以及第四FFCC模型进行计算处理，分别得到该图像的第一调节值、第二调节值、第三调节值和第四调节值。然后，将这四个调节值输入卡尔曼滤波器进行计算，得到该图像的调节值。通过所述AI自动白平衡算法得到的所述图像调节值是一个准确度极高的调节值，当图像信号处理器(ISP)使用所述图像的调节值来对该图像的色彩进行调节时，能够很好地修正图像的色差，令图像的白平衡效果更好。平衡效果更好。平衡效果更好。

【技术实现步骤摘要】
一种AI自动白平衡的算法和电子设备


[0001]本申请涉及图像处理领域，尤其涉及一种AI自动白平衡的算法和电子设备。

技术介绍

[0002]由于人眼具有独特的适应性，使我们有的时候不能发现色温的变化。比如在钨丝灯下呆久了，并不会觉得钨丝灯下的白纸偏红，如果突然把日光灯改为钨丝灯照明，就会觉查到白纸的颜色偏红了，但这种感觉也只能够持续一会儿。摄像机的电荷耦合元件电路(Charge
‑
coupled Device，CCD)或CMOS电路没有办法像人眼一样会对光源的颜色变化进行修正，所以如果摄像机的白平衡设置同景物照明的色温不一致就会发生偏色。因此，如何提高白平衡的准确性是技术人员日益关注的问题。

技术实现思路

[0003]本申请实施例提供了一种AI自动白平衡的算法和电子设备，解决了通过传统自动白平衡算法得到的图像光源的RGB_GAIN(图像调节值)准确度不高，进而造成电子设备基于所述RGB_GAIN对图像进行白平衡调节效果不佳的问题。
[0004]第一方面，本申请实施例提供了一种AI自动白平衡的算法，包括：本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种AI自动白平衡的算法，其特征在于，包括：将第一图像分别输入第一FFCC模型、第二FFCC模型、第三FFCC模型和第四FFCC模型进行计算，得到第一调节值、第二调节值、第三调节值和第四调节值，所述第一调节值通过所述第一FFCC模型得到的，所述第二调节值通过所述第二FFCC模型得到的，所述第三调节值通过所述第三FFCC模型得到的，所述第四调节值通过所述第四FFCC模型得到的；将所述第一调节值、所述第二调节值、所述第三调节值和所述第四调节值输入卡尔曼滤波器进行计算，得到所述第一图像的调节值，所述第一图像的调节值用于调节所述第一图像的颜色；其中，所述第一FFCC模型用于标识图像亮度值大于第一阈值对应的模型，所述第二FFCC模型用于标识图像亮度值小于或等于第一阈值对应的模型，所述第三FFCC模型用于标识图像光源的D
uv
大于第二阈值对应的模型，所述第四FFCC模型用于标识图像光源的D
uv
小于或等于第二阈值对应的模型。2.如权利要求1所述的方法，其特征在于，所述将第一图像分别输入第一FFCC模型、第二FFCC模型、第三FFCC模型和第四FFCC模型进行计算，得到第一调节值、第二调节值、第三调节值和第四调节值，具体包括：将所述第一图像分别输入所述第一FFCC模型、所述第二FFCC模型、所述第三FFCC模型和所述第四FFCC模型进行计算，得到第一色度坐标和第一协方差矩阵、第二色度坐标和第二协方差矩阵、第三色度坐标和第三协方差矩阵、第四色度坐标和第四协方差矩阵；根据所述第一图像的亮度值计算所述第一FFCC模型的权重值f1和所述第二FFCC模型的权重值f2；基于所述f1、所述第一协方差矩阵和所述第一色度坐标，计算得到所述第一调节值；基于所述f2、所述第二协方差矩阵和所述第二色度坐标，计算得到所述第二调节值；基于所述第一调节值和所述第二调节值计算得到所述第一图像光源的D
uv
；根据所述第一图像光源的D
uv
计算所述第三FFCC模型的权重值f3和所述第四FFCC模型的权重值f4；基于所述f3、所述第三协方差矩阵和所述第三色度坐标，计算得到所述第三调节值；基于所述f4、所述第四协方差矩阵和所述第四色度坐标，计算得到所述第四调节值。3.如权利要求2所述的方法，其特征在于，所述根据所述第一图像的亮度值计算所述第一FFCC模型的权重值f1和所述第二FFCC模型的权重值f2，具体包括：根据公式得到所述权重值f1；根据公式f2＝1
‑
f1，得到所述权重值f2；其中，所述Lv
thres
为所述第一阈值，所述x为所述第一图像的亮度值，所述Lv
mult
用于表征所述x在所述f1上的变化速率。4.如权利要求2
‑
3任一项所述的方法，其特征在于，所述基于所述f1、所述第一协方差矩阵和所述第一色度坐标，计算得到所述第一调节值，具体包括：将所述第一协方差矩阵根据公式计算，得到更新后的第一协方差矩阵，所述Sigma1为所述第一协方差矩阵，所述Sigma
′1为所述更新后的第一协方差
矩阵；将所述第一色度坐标和所述Sigma
′1根据公式Mu
′1＝Mu1*(Sigma
′1)
‑1计算，得到所述第一调节值；其中，所述Mu1为所述第一色度坐标，所述Mu
′1为所述第一调节值。5.如权利要求2
‑
3任一项所述的方法，其特征在于，所述基于所述f2、所述第二协方差矩阵和所述第二色度坐标，计算得到所述第二调节值，具体包括：将所述第二协方差矩阵根据公式计算，得到更新后的第二协方差矩阵，所述Sigma2为所述第二协方差矩阵，所述Sigma
′2为所述更新后的第二协方差矩阵；将所述第二色度坐标和所述Sigma
′2根据公式Mu
′2＝Mu2*(Sigma
′2)
‑1计算，得到所述第二调节值；其中，所述Mu2为所述第二色度坐标，所述Mu
′2为所述第二调节值。6.如权利要求4
‑
5任一项所述的方法，其特征在于，所述基于所述第一调节值和所述第二调节值计算得到所述第一图像光源的D
uv
，具体包括：根据公式Mu
′
＝Mu
′1*(Sigma
′1)
‑1+Mu
′2*(Sigma
′2)
‑1计算，得到融合后的色度坐标，所述Mu
′
为所述融合后...

【专利技术属性】
技术研发人员：钱彦霖，郗东苗，金萌，罗钢，朱聪超，
申请(专利权)人：荣耀终端有限公司，
类型：发明
国别省市：