【VS开发】【图像处理】相机中白平衡的算法模拟实现
相机主要技术点为3A算法。
而3A算法主要指的是自动对焦(AF)、自动曝光(AE)及自动白平衡(AWB)。自动白平衡:根据光源条件调整图片颜色的保真程度。
网上时常有类似招聘如下的招聘信息:
---------------------------------------------- ----------------------------------------------
Camera/ISP 算法工程师摄像机3A算法软件工程师
这里随机摘录一些具体要求。
任职要求：
1、本科以上学历，天文，物理，机电、工业自动化，电子相关专业，硕士学历优先考虑；
2、本科毕业3年以上，硕士毕业1年以上的相关行业相关工作经验要求；
3、熟练掌握C/C++或者FPGA 开发语言,数据结构,MATLAB,信号和系统；
4、掌握数字色度学，数字图像处理，数字影像处理的基本知识；
5、熟悉摄像机成像原理；
6、掌握3A(AF,AE,AWB)算法之一；
7、对于自动化控制，数字信号采样，滤波，负反馈，PID算法有实际经验；
8、理解从镜头到SENSOR,电机，ISP，编码器，采集，显示通道一些列变化。
任职要求：
1. 精通camera的3A（AE,AWB,AF）算法原理和设计思路， 有3A算法的设计经验为佳
2. 具备丰富ISP（图象处理器） 开发经验，熟悉MTK，QUALCOMM, OV等便携式终端上应用的ISP开发环境。有上述环境下开发经验为佳。
3. 精通数字图像处理原理和基础知识。
4. 熟悉C/C++语言，有开发经验为佳
5. 有手机/便携式相机3A算法实现/应用经验
6. 精通CMOS sensor的工作原理
---------------------------------------------- ----------------------------------------------
而这类职位一般都是高薪待遇。
然后问题来了，市面上3A算法相关资料都非常稀少，就连相关书籍都很少提及算法细节，而他们基本上都会要求精通3A算法至少之一。
而关于白平衡算法，比较不错的资料是这份：
基于灰度世界、完美反射、动态阈值等图像自动白平衡算法的原理、实现及效果
之前多次与博主laviewpbt探讨相关的知识，受益匪浅。
而据我所知，绝大多数的相机采用的基础算法便是灰度世界算法,然后在这算法的基础上再改进。
贴一下《 基于灰度世界、完美反射、动态阈值等图像自动白平衡算法的原理、实现及效果 》灰度世界法的大概内容。
---------------------------------------------- ----------------------------------------------
灰度世界算法（Gray World)
是以灰度世界假设为基础的,该假设认为对于一幅有着大量色彩变化的图像, R、 G、 B 三个分量的平均值趋于同一个灰度K。一般有两种方法来确定该灰度。
（1)直接给定为固定值, 取其各通道最大值的一半,即取为127或128；
（2)令 K = (Raver+Gaver+Baver)/3,其中Raver,Gaver,Baver分别表示红、 绿、 蓝三个通道的平均值。
算法的第二步是分别计算各通道的增益：
Kr=K/Raver;
Kg=K/Gaver;
Kb=K/Baver;
算法第三步为根据Von Kries 对角模型,对于图像中的每个像素R、G、B，计算其结果值：
Rnew = R * Kr;
Gnew = G * Kg;
Bnew = B * Kb;
对于上式，计算中可能会存在溢出（>255,不会出现小于0的)现象，处理方式有两种。
a、 直接将像素设置为255，这可能会造成图像整体偏白。
b、 计算所有Rnew、Gnew、Bnew的最大值，然后利用该最大值将将计算后数据重新线性映射到[0,255]内。实践证明这种方式将会使图像整体偏暗，建议采用第一种方案。
---------------------------------------------- ----------------------------------------------
算法的大概思路就是评估一张图片RGB三个通道的中最能表达该通道富含信息的值，然后以该值为基准重新调整像素。
这样就会存在评估不够准确的问题，导致各通道像素信息差距过大，形成噪点以及偏色等现象。
而laviewpbt在博文中写道，建议采用第一种方案，这个我认为也不妥，因为如果采用取最大值的方案就会导致在特定情况明显不均衡，例如该通道大多数的值落在最小值周围，而却存在一个遥远处的最大值，那么就会导致像素信息差距过大，就很糟糕了。
所以在第二种思路上进行进一步改进比较稳妥，因为可用的信息比较多，不容易出问题。
第二种思路，最简单的另一种改进就是采用灰度法。
均值法: K = (Raver+Gaver+Baver)/3
我们知道常用的视频采集编码是YUV。
YUV相关见百度百科: YUV
其中的Y为:
Y =0.299*R + 0.587*G+0.114*B
故灰度法相应可对应为:
K=0.299*Raver + 0.587*Gaver+0.114*Baver
经过实测，这样的处理后效果还不错。
贴上对比图：
原图
均值法
灰度法
单从肉眼上去分辨两张图片,的确很难分出优劣。
不过我也只是大概点一下这个思路而已，有所积累的人，看到这，应该可以发散出更多的想法。
接下来我要说的是具体相机中的钨丝灯等手动白平衡是如何实现的。
简单的说就是色温条件。
那么基于灰度世界这个白平衡算法可以怎么实现这种调节呢？！
这里贴出简单实现的C代码：
switch (preset) { case AUTO: Raver = (SumR / numberOfPixels); Gaver = (SumG / numberOfPixels); Baver = (SumB / numberOfPixels); break; case CLOUDY: Raver = (SumR *1.953125 / numberOfPixels); Gaver = (SumG*1.0390625 / numberOfPixels); Baver = (SumB / numberOfPixels); break; case DAYLIGHT: Raver = (SumR *1.2734375 / numberOfPixels); Gaver = (SumG / numberOfPixels); Baver = (SumB*1.0625 / numberOfPixels); break; case INCANDESCENCE: Raver = (SumR *1.2890625 / numberOfPixels); Gaver = (SumG / numberOfPixels); Baver = (SumB*1.0625 / numberOfPixels); break; case FLUORESCENT: Raver = (SumR *1.1875 / numberOfPixels); Gaver = (SumG / numberOfPixels); Baver = (SumB*1.3125 / numberOfPixels); break; case TUNGSTEN: Raver = (SumR / numberOfPixels); Gaver = (SumG*1.0078125 / numberOfPixels); Baver = (SumB*1.28125 / numberOfPixels); break; default: break; }
enum WB_PRESET{ //自动白平衡 AUTO, //阴天 7500k CLOUDY, //日光 6500k DAYLIGHT, //白热光 5000k INCANDESCENCE, //日光灯 4400k FLUORESCENT, //钨丝灯 2800k TUNGSTEN, };
阴天
日光
白热光
日光灯
钨丝灯
这里只是起到一个演示作用，具体的参数，可按实际需求酌情进行修改。
