074、自动白平衡（AWB）统计法进阶：色温曲线拟合与灰区自适应调整

074、自动白平衡AWB统计法进阶色温曲线拟合与灰区自适应调整一、一个让我熬夜三天的bug去年做某旗舰机项目实验室里拍白炽灯场景AWB输出色温2700K看起来挺准。结果产线同事反馈同一批模组在产线暗室用标准D65光源测试有20%的机器AWB偏绿Delta C达到8以上。我盯着log看了三天发现灰区统计点的分布在不同模组间差异巨大——有的集中在低色温区有的往高色温飘。问题出在我们用的灰区阈值是固定的而sensor的响应特性、镜头的红外截止深度、甚至模组组装时的微透镜偏移都会让“灰点”的分布产生系统性偏移。这个案例说明AWB统计法不是简单的“找灰点→算色比→查表”灰区自适应调整和色温曲线拟合才是工程落地的关键。二、灰区定义别被“灰色”骗了很多人以为灰区就是RGB的区域实际调试中根本不是这样。sensor的RGB响应受QE曲线、镜头透过率、IR cut filter的截止波长影响即使拍标准灰卡R/G和B/G也不是1.0。我们通常用色比空间R/G, B/G来定义灰区但阈值怎么设早期我踩过一个坑直接用固定矩形框比如R/G在0.81.2B/G在0.81.2。结果在低色温下R/G可能到1.8B/G只有0.4灰点全被框在外面AWB直接崩了。后来改用色温相关的动态阈值但动态阈值怎么调靠查表表做细了不同sensor要重新标定表做粗了场景一换就翻车。我的做法用色温曲线拟合来定义灰区。先标定一组标准色温下的灰点D65、D50、A光源、TL84等这些点连起来就是一条“灰点轨迹”。灰区就是这条轨迹的邻域——邻域宽度取决于sensor噪声水平和模组一致性。邻域形状不是圆形而是沿着轨迹方向拉长的椭圆因为色温变化时R/G和B/G的耦合关系不是各向同性的。三、色温曲线拟合别用多项式硬怼色温曲线本质是黑体辐射轨迹在色比空间的投影。理论上可以用Planckian locus公式但实际sensor的响应是非线性的加上镜头色差直接套公式误差很大。我试过多项式拟合比如R/G a0 a1T a2T^2在低色温段拟合残差小但到了高色温6500K就飘了。后来改用分段样条插值节点选在2800K、4000K、5500K、6500K、8000K每段用三次样条。这样拟合残差控制在1%以内但计算量大了——嵌入式平台跑样条插值每帧要算几百个点CPU扛不住。工程妥协离线标定时用样条插值生成密集的色温-色比查找表比如每100K一个点在线运行时用线性插值查表。查表时注意边界处理——别让色温超出标定范围否则外推误差会爆炸。我见过有人把色温推到12000K查表直接返回一个离谱的色比AWB瞬间偏紫。四、灰区自适应调整从“死区”到“活区”固定灰区的问题在于不同场景下有效灰点的分布密度差异巨大。比如室内暖光场景灰点集中在低色温区户外阴天灰点分布在高色温区。如果灰区形状固定低色温场景下高色温区的灰点统计权重被浪费反之亦然。自适应策略根据当前帧的统计结果动态调整灰区的中心位置和宽度。中心位置调整用上一帧的AWB输出色温作为参考将灰区中心沿色温曲线移动到对应色温点。这样当前帧的灰点统计会集中在真实色温附近减少噪声干扰。宽度调整宽度与场景的“色彩丰富度”相关。如果场景中灰点分布很集中比如白墙宽度可以收窄提高色温估计精度如果场景色彩丰富比如花丛灰点分布散宽度要放宽避免漏掉有效灰点。我用的指标是“灰点密度熵”——熵值大说明分布均匀宽度放宽熵值小说明分布集中宽度收窄。这里踩过坑宽度调整不能太快否则AWB会振荡。比如场景从白墙切到花丛灰点密度熵瞬间变大宽度突然放宽下一帧统计到更多非灰点色温估计跳变再下一帧宽度又收窄……形成正反馈振荡。解决办法对宽度做低通滤波时间常数设到0.5秒左右。五、代码实现别这样写// 错误示范固定灰区阈值#defineGRAY_RG_MIN0.8f#defineGRAY_RG_MAX1.2f#defineGRAY_BG_MIN0.8f#defineGRAY_BG_MAX1.2f// 这样写低色温场景直接翻车// 正确做法基于色温曲线的动态灰区typedefstruct{floatrg;// R/Gfloatbg;// B/Gfloatweight;// 统计权重}gray_point_t;// 色温曲线查表离线标定生成staticconstfloatawb_curve[][3]{{2800.0f,1.85f,0.42f},// 色温, R/G, B/G{4000.0f,1.45f,0.65f},{5500.0f,1.20f,0.85f},{6500.0f,1.10f,0.95f},{8000.0f,1.02f,1.05f},};// 根据色温查表得到灰区中心voidget_gray_center(floatcct,float*rg_center,float*bg_center){// 线性插值注意边界处理if(cctawb_curve[0][0]){*rg_centerawb_curve[0][1];*bg_centerawb_curve[0][2];}elseif(cctawb_curve[4][0]){*rg_centerawb_curve[4][1];*bg_centerawb_curve[4][2];}else{// 找到区间线性插值for(inti0;i4;i){if(cctawb_curve[i][0]cctawb_curve[i1][0]){floatt(cct-awb_curve[i][0])/(awb_curve[i1][0]-awb_curve[i][0]);*rg_centerawb_curve[i][1]t*(awb_curve[i1][1]-awb_curve[i][1]);*bg_centerawb_curve[i][2]t*(awb_curve[i1][2]-awb_curve[i][2]);break;}}}}// 判断是否为灰点带自适应宽度intis_gray_point(gray_point_t*pt,floatcct_ref,floatwidth_factor){floatrg_center,bg_center;get_gray_center(cct_ref,rg_center,bg_center);// 宽度沿轨迹方向拉伸垂直方向压缩// 这里用椭圆距离半轴长度与width_factor相关floatdrgpt-rg-rg_center;floatdbgpt-bg-bg_center;floatdistsqrtf(drg*drgdbg*dbg);// width_factor由场景熵决定范围0.5~2.0floatthreshold0.05f*width_factor;// 基础阈值0.05可调return(distthreshold);}别这样写把width_factor设成固定值或者忘了对cct_ref做限幅。cct_ref如果来自上一帧AWB输出在场景切换时可能跳变导致灰区突然移动统计失效。建议对cct_ref做时间滤波或者用多帧中值。六、调试经验那些文档里不会写的事灰区形状的“胖瘦”与sensor噪声的关系高ISO下噪声大灰点分布散灰区要“胖”一点低ISO下噪声小灰区可以“瘦”。我见过有人把灰区宽度设成ISO的线性函数但实际噪声是平方根关系用sqrt(ISO)更合理。色温曲线标定的“陷阱”标定时用标准光源灰卡但实际场景中物体反射率不是中性的。比如拍绿色植物灰点统计会偏向绿色方向导致AWB偏品。解决办法在灰区判断时加入“色彩饱和度”约束——饱和度太高的点直接排除不管它离灰区多近。产线一致性补偿不同模组的灰点轨迹有偏移可以在产线校准阶段记录每个模组的偏移量写入OTP。AWB运行时读取偏移量对色温曲线做平移。这个偏移量通常很小R/G偏差0.02但足以让Delta C从8降到2。极端场景的兜底当场景中几乎没有灰点比如纯色背景灰区统计会失效。这时要回退到“全局色温估计”——用整帧的RGB均值做粗略估计或者直接锁定上一帧色温。我习惯在灰点数量低于总像素的1%时触发回退。七、个人建议AWB统计法做到这个程度已经能覆盖90%的场景。但剩下的10%——比如舞台灯光下混合色温、水下拍摄、荧光灯频闪场景——需要结合机器学习或者多帧融合。不要试图用一个算法解决所有问题工程上要学会“场景识别策略切换”。比如检测到频闪时关闭AWB的帧间平滑改用单帧快速响应检测到混合色温时允许AWB输出两个色温值让ISP做局部白平衡。最后说一句调试AWB别只看实验室数据。去产线蹲一天看看不同模组在真实光源下的表现比看一百篇论文都有用。