前言
使用灰度世界法在fpga实现白平衡
注意:后续技术分享,第一时间更新,以及缩放、白平衡及各种滤波算法完整代码,将在VX群发布,请添加VX:Zeus20010223
一、白平衡是什么?
白平衡(White Balance),或者称之为色彩恒常性(Color Constancy),即在光照变化的情况下保持景物色彩一致的能力。人类大脑拥有有关色彩恒常性的相关机制,可以使同一个物体(通常认为自然界中物品大多数不会变色,改变的是光源)在不同的光源条件下都被大脑识别为近似的颜色。而相机实际上是利用内部复杂的图像处理管线(image processing pipeline),将进入传感器的光强(准确来说是辐照度irradiance)变换到像素值。而由于不同的光源条件,导致同一个物体反射的光线会有所差别,这也导致了最终相机会认为同一个物品在不同光线条件下为不同的颜色。为了解决这一问题,现代相引入了白平衡调整的功能。
使用灰度世界法来实现白平衡
灰度世界法(Gray World)是以灰度世界假设为基础的,该假设认为对于一幅有着大量色彩变化的图像, R、G、B 三个分量的平均值趋于同一个灰度K。一般有两种方法来确定该灰度:
(1)直接给定为固定值, 取其各通道最大值的一半,即取为128;
(2)令K = (Raver+Gaver+Baver)/3,其中Raver,Gaver,Baver分别表示红、绿、蓝三个通道的平均值。
算法的第二步是分别计算各通道的增益:
Kr=K/Raver;
Kg=K/Gaver;
Kb=K/Baver;
算法第三步对于图像中的每个像素R、G、B计算其结果值:
Rnew = R * Kr;
Gnew = G * Kg;
Bnew = B * Kb;
举个例子那么Rnew的和 = R1 * Kr + R2Kr + ……RnKr = R1K/Raver + R2K/Raver + ……RnK/Raver; Rnew和的平均值 = (R1K/Raver + R2K/Raver + ……RnK/Raver)/n =( R1 /n +R2 /n + ……Rn /n) K/Raver = RaverK/ Raver = K;
对于上式,计算中可能会存在溢出(>255)现象,直接将像素设置为255。
二、fpga实现
直接取k值为128
对于每个通道的平均值需要累加之后除像素 假设使用1280*720
由于综合时会向下取整 所以需要进行特殊处理 否则均值会直接为0
localparam [63:0] total_recip = (2**31)/(COL*ROW);//1/(COL*ROW)
将除数扩大2的31次方倍 后续通过截位操作还原
通过查找表的形式将有范围的变量转换为常量
recip u1_recip(
.ava (accu_r_ava[38:31] ),
.recip (recip_r )
);
recip u2_recip(
.ava (accu_g_ava[38:31] ),
.recip (recip_g )
);
recip u3_recip(
.ava (accu_b_ava[38:31] ),
.recip (recip_b )
);
由于白平衡是根据当前像素的值来对下一帧进行操作 所以第一帧不进行处理
always@(posedge clk)
if(!rst_n)
begin
awb_r <= 'd0;
awb_g <= 'd0;
awb_b <= 'd0;
end
else
begin
awb_r <= gain_r*pre_data[23:16];
awb_g <= gain_g*pre_data[15:8];
awb_b <= gain_b*pre_data[7:0];
end
三、仿真验证
处理结果
结果验证
可以看到Rgain增益值误差不大
总结
主要介绍了在fpga实现时的两个操作