基于多尺度wls滤波的雾天图像恢复方法

基于多尺度wls滤波的雾天图像恢复方法
【专利摘要】本发明属于图像处理领域，涉及基于多尺度WLS滤波的雾天图像恢复方法。所述方法包括：根据暗通道先验原理估计大气光照强度A；结合得到的大气光照强度A的值，对图像I进行白平衡，并简化大气散射模型；根据雾天环境的物理属性，对大气耗散函数V进行粗估计，得到；采用多尺度WLS滤波对大气耗散函数进行细估计，得到Vi；结合获得的多尺度大气耗散函数Vi，根据简化大气散射模型恢复雾天图像为理想天气下的图像Ri；结合获得的多尺度恢复图像Ri，采用色调映射算法调整图像动态范围，进行可视度的提升，得到最终结果R。本发明能够有效提高雾天图像的对比度与清晰度，具有计算复杂度较低、执行效率较强等优点。
【专利说明】基于多尺度WLS滤波的雾天图像恢复方法

【技术领域】
[0001] 本发明属于图像处理领域，具体涉及一种基于多尺度WLS(Weightedleast squares,加权最小二乘法）滤波的雾天图像处理方法。

【背景技术】
[0002] 在雾、霾等天气条件下，由于场景的能见度降低，图像中目标对比度和颜色等特征 被衰减，这极大地限制和影响了户外工作系统，如视频监控、地形勘测、自动驾驶等。因此， 对雾天图像恢复的实现是一个极具现实意义的问题。
[0003] 目前，对雾天图像进行清晰化处理的方法大体上有两类：基于模型的和非模型的 算法。这里的模型指的是大气散射模型，它是一种描述光在大气中传输的物理模型。其中， 基于非模型的算法并不要求了解图像退化的原因，只能在一定程度上改善视觉效果。此类 方法属于图像增强技术范畴，主要包括：直方图方法，基于小波的方法，以及基于Retinex 的算法等；基于模型的算法通过了解图像退化的内在原因而进行逆运算来改善图像的可视 度，采用该类方法处理图像属于图像恢复技术范畴，由于它们利用了图像退化的物理机理， 因此这类算法更加可靠，去雾效果更加明显。
[0004] 近年来，基于模型的研究取得了较大的进展，特别是针对单幅图像去雾技术取得 了重大突破° 文献I(TarelJP,HautiN.Fastvisibilityrestorationfromasingle colororgraylevelimage[J].ProceedingsofIEEEConferenceonInternational ConferenceonComputerVision,Kyoto,Japan, 2009:20-28)假设大气耗散函在可行域 中逼近最大值，且局部变化平缓，提出了一种快速图像去雾算法。该方法采用中值滤波方 法估计大气耗散函数，其缺点是细节保持能力较差。文献2(HeK，SunJ，TangX.Single imagehazeremovalusingdarkchannelprior[C].ProceedingsofIEEEConference onComputerVisionandPatternRecognition(CVPR).NewYork,USA:IEEEComputer Society，2009:1956-1963)通过对户外无雾图像的统计实验提出了暗原色先验理论，利用 该理论获得了很好的去雾效果，并能得到相应的深度图像。该方法具备物理有效性，但算法 复杂度高、处理速度慢。


【发明内容】

[0005] 本发明的目的在于提出一种基于多尺度WLS滤波的雾天图像恢复方法，采用大气 散射模型对雾天场景进行建模，准确求解大气耗散函数，有效地去除图像中雾气的影响，改 善雾天图像可视度。
[0006] 下面给出本发明所述方法的原理。
[0007] 在大雾天气情况下，大气中漂浮的悬浮颗粒的直径比较大，它不仅对入射光有直 接的影响，也会反射周围的环境光。大气散射模型能够用于描述这种光在雾天传输的物理 特性，该模型表示如下：
[0008] I(x，y) =R(x，y)e-0d(x'y)+A(l-e-ed(x'y))
[0009] 其中，I为原图像亮度，R为理想天气下的清晰图像亮度，A为全局性的大气光照， 3是大气散射系数，雾天环境下它可认为是一个与波长无关的常量，d为景深。本发明为了 避免直接求解d和3两个参数，采用求解大气耗散函数的方式，其表示为：
[0010] V(x，y) =l-e-0d(x'y)
[0011] 其中，V为大气耗散函数，包含景深信息，能够粗略表示雾气的影响。结合上式与 大气散射模型可知，去雾操作过程转变为通过I计算A和V两个参数，进而求解清晰图像R。 R也可以分解为：
[0012] R(x,y) =Ap(x,y)
[0013] 其中，P为场景反照率。
[0014] 根据上述原理，本发明所述方法包括以下步骤：
[0015] 步骤1 :根据暗通道先验原理估计大气光照强度A。
[0016] 步骤2 :结合得到的大气光照强度A的值，对图像I进行白平衡，并简化大气散射 模型。
[0017] 步骤3 :根据雾天环境的物理属性，对大气耗散函数进行粗估计，得到(？。
[0018] 步骤4:采用多尺度WLS滤波对大气耗散函数f进行细估计，得到Vi,i为大于零的 整数。
[0019] 步骤5:结合获得的多尺度大气耗散函数Vi,根据简化大气散射模型恢复雾天图像 为理想天气下的图像氏。
[0020] 步骤6 :结合获得的多尺度恢复图像Ri,采用色调映射算法调整输出图像动态范 围，进行可视度的提升，得到最终结果R。
[0021] 与现有技术相比，本发明具有以下优点：
[0022] (1)本发明能够有效提高雾天图像的对比度与清晰度，具有计算复杂度较低、执行 效率较强等优点。实验证明，文献1所述方法去雾处理后有校正过度的现象，且存在雾气去 除不完全现象。文献2所述方法的去雾效果较为自然，但是细节凸显能力较差，执行耗时较 长。相比较二者，本发明所述方法去雾效果显著，较好地保持了图像颜色，增强了细节信息， 大幅度提高了图像对比度。
[0023] (2)本发明在许多领域中具有重要的应用价值。比如，对于灰度图像和彩色图像均 能处理，本发明能够满足视频监控、目标跟踪、信息识别等对于图像可视度的要求。另外，本 发明在高速公路交通监控、军事侦察、遥感探测等领域具有很高的实用价值。

【专利附图】

【附图说明】
[0024] 图1为本发明所述方法的流程图；
[0025] 图2为本发明实施例获得的结果：(a)原图像，（b)单尺度细化结果V1, (c)单尺度 去雾结果R1, (d)实验最终结果R;
[0026] 图3?5为本发明与现有技术效果对比1?3,（a)原图像，（b)应用文献1得到 的结果，（c)应用文献2得到的结果，（d)应用本发明得到的结果。

【具体实施方式】
[0027] 下面结合附图和实施方式对本发明做进一步说明。
[0028] 本发明所述方法的流程图如图1所示，具体包括以下步骤：
[0029] 步骤1 :输入原始图像，以图2(a)为例，该图为大小为315X315的雾天图像，采用 基于暗通道先验的方法求解全局常量大气光照A。
[0030] 暗通道先验认为在非天空图像块中，至少有一个颜色通道在某些像素上有很低的 亮度。对于一幅雾天图像I，可以定义暗通道为：
[0031]

【权利要求】
1. 基于多尺度WLS滤波的雾天图像恢复方法，其特征在于包括以下步骤： 步骤1 :根据暗通道先验原理估计大气光照强度A; 步骤2 :结合得到的大气光照强度A的值，对图像I进行白平衡，并简化大气散射模型； 步骤3 :根据雾天环境的物理属性，对大气耗散函数V进行粗估计，得到P; 步骤4 :采用多尺度加权最小二乘法WLS滤波对大气耗散函数进行细估计，得到Vi,i为 大于零的整数； 步骤5:结合获得的多尺度大气耗散函数Vi,根据简化大气散射模型恢复雾天图像为理 想天气下的图像Ri ; 步骤6:结合获得的多尺度恢复图像Ri,采用色调映射算法调整图像动态范围，进行可 视度的提升，得到最终结果R。
2. 根据权利要求1所述的基于多尺度WLS滤波的雾天图像恢复方法，其特征在于，步骤 1所述大气光照强度A的估算方法如下： 暗通道先验认为在非天空图像块中，至少有一个颜色通道在某些像素上有很低的亮 度；对于一幅雾天图像I，定义暗通道为： I'1'"'1'(q)=min fmin(/r (/;))] 其中，IdaA为I的暗通道，c为颜色通道索引值，Γ是I的一个颜色通道，Ω是以像素q为中心的一个局部块；根据暗通道先验理论可知，如果I是一幅无雾的户外图像，则除了天 空区域，IdaA的亮度都较低而且接近0,即位于雾天图像中的清晰区域产k -O; 将暗通道IdaA中的像素按亮度值从大到小排序，选取排序后亮度值的前0. 1 %的像素， 然后，将其对应的原图像I中相同位置的像素表示为1〇.1%，这些点都是雾最浓的点，认为大 气光照值位于Itl. 1%范围中；为了去雾处理更加彻底，取Ia1%的均值Am_ =mean(Ia1%)作为 大气光照的初始值；由于雾天环境常常伴随着阴天，致使雾气偏离了纯白颜色，为此对A_n 进行校正，将其映射接近白色（1，1，1)，即： cgRXj'B 其中，A即为所要求解的大气光照值。
3. 根据权利要求1所述的基于多尺度WLS滤波的雾天图像恢复方法，其特征在于，步骤 2所述对图像I进行白平衡，并简化大气散射模型的方法如下： 原大气散射模型表不为： I(x，y) =R(X，y)e-ed(x'y)+A(l-e-ed(x'y)) 其中，I为原图像亮度，R为理想天气下的清晰图像亮度，A为全局性的大气光照，β是 大气散射系数，雾天环境下它可认为是一个与波长无关的常量，d为景深；为了避免直接求 解d和β两个参数，采用求解大气耗散函数的方式，其表示为： V(x，y) =l-e-0d(x'y) 其中，V为大气耗散函数，包含景深信息，能够粗略表示雾气的影响； 通过白平衡处理后得到的简化大气散射模型表示为： Γ(x.y) =I(x,y)/A=P(x,y) (1-V(x,y))+V(x,y) 其中，I'为白平衡处理后的图像，P为场景反照率；且： R(x，y) =Ap(X，y)。
4. 根据权利要求I所述的基于多尺度WLS滤波的雾天图像恢复方法，其特征在于，步骤 3所述根据雾天环境的物理属性，对大气耗散函数V进行粗估计的方法如下： 根据雾天场景的物理属性，V需要满足下面2个约束条件： (1) 0V(x,y) ^ 1 (2)V(x，y)不大于Γ(x，y)的最小颜色分量 通过上述约束，将白平衡后的雾天图像的每个像素在RGB三个通道的最小颜色分量定 义为大气耗散函数的粗估计结果f，即： F(x，V) =min(7'(.?.V)) -ce{R,GM\ " 对于一幅灰度图，则有丨αυ)="x,_v：h此时的f不具备大气耗散函数局部平滑等特 性，需要进一步的细化。
5. 根据权利要求1所述的基于多尺度WLS滤波的雾天图像恢复方法，其特征在于，步骤 4所述采用WLS滤波对大气耗散函数f进行细估计的方法如下： WLS滤波表示为最小化如下的代价函数：
其中，公式左端第一项为数据项，该项的目标是最小化输入图像f和输出图像V的差 异；第二项为平滑项目，该项通过最小化V的X与y方向的偏导数来达到平滑目的，λi为 调节参数，i为尺度索引值；g为平滑权值，定义为：
其中，ε为一个很小的常量，防止除数为零，Yi为调节梯度变化的敏感程度参数；将WLS的最小能量公式离散化后转化成矩阵形式： (V -Vf (V - V) +D[ADV+V1 D1v AvDV) 其中，Ax和Ay是关于细化权值gx和gy的对角矩阵；Dx和Dy是浐的X和y方向的前向 差分矩阵，而Df和D17为后向差分矩阵，上式的最小化可以转化为以下的公式求解： (Ε+λβψ=V 其中，E表示单位矩阵，G= +Z>y,.为稀疏Laplacian矩阵；上式可以通过线 性求解获得细化处理后的Vi ;采用不同λi和Yi参数进行多尺度大气耗散函数求解，当尺 度数量i= 3时，获得的结果已经较为理想，当1>3时，视觉效果提升程度较小，且增加了计 算复杂度，故将尺度数量设定为3,即i= 3。
6. 根据权利要求1?5中任何一项所述的基于多尺度WLS滤波的雾天图像恢复方法， 其特征在于，步骤5所述恢复雾天图像为理想天气下的图像Ri的方法如下： 在求取大气光强度A与大气耗散函数Vi的基础上，直接恢复出场景在理想天气条件下 的图像亮度： n . 、I(x.v)-kV(x.v), R(x,v) = ·' ·-'' ·xA " \-V：(x,y) 其中，k为恢复调节参数，0〈k〈l，用于控制去雾的程度。
7.根据权利要求6所述的基于多尺度WLS滤波的雾天图像恢复方法，其特征在于，步骤 6所述调整输出图像动态范围，进行可视度提升的方法如下： 首先进行颜色空间的转换，将图像Ri由RGB颜色空间转换到CIELAB颜色空间，仅对图 像的亮度空间L进行处理；假设R1的雾天图像恢复效果最好，则设b作为基础层，即b=R1, 设两个细节层d1与d2,即d1 =R1-R2,d2 =R2-R3 ;则色调映射公式定义为： Rp -μ+€(δα,φρ -/〇+C(^, 其中，P表示像素，μ为图像亮度范围的均值，c为s型调节函数，用于调节动态范围， 即C(a，u) =l/(l+exp(-au)), η为曝光参数，δ。，61与δ2为增益参数； 经过上述的处理后，将图像由CIELAB颜色空间转换到RGB颜色空间，即可得到最后的 雾天图像恢复结果R。
【文档编号】G06T5/00GK104318528SQ201410562140
【公开日】2015年1月28日 申请日期:2014年10月21日 优先权日:2014年10月21日
【发明者】赵宏宇, 肖创柏, 禹晶, 段娟 申请人:北京工业大学