一种基于自适应大气光和加权引导滤波的夜间图像去雾算法

赵波 李洪平 金汝宁 唐万松 刘相宜

赵波, 李洪平, 金汝宁, 等. 一种基于自适应大气光和加权引导滤波的夜间图像去雾算法 [J]. 工程科学与技术, 2023, 55(3): 225-234. doi: 10.15961/j.jsuese.202101218
引用本文: 赵波, 李洪平, 金汝宁, 等. 一种基于自适应大气光和加权引导滤波的夜间图像去雾算法 [J]. 工程科学与技术, 2023, 55(3): 225-234. doi: 10.15961/j.jsuese.202101218
ZHAO Bo, LI Hongping, JIN Runing, et al. A Night Image Dehazing Algorithm Based on the Adaptive Atmospheric Light and the Weighted Guided Filter [J]. Advanced Engineering Sciences, 2023, 55(3): 225-234. doi: 10.15961/j.jsuese.202101218
Citation: ZHAO Bo, LI Hongping, JIN Runing, et al. A Night Image Dehazing Algorithm Based on the Adaptive Atmospheric Light and the Weighted Guided Filter [J]. Advanced Engineering Sciences, 2023, 55(3): 225-234. doi: 10.15961/j.jsuese.202101218

一种基于自适应大气光和加权引导滤波的夜间图像去雾算法

基金项目: 四川省重大科技专项(2020YFSY0058)
详细信息
    • 收稿日期:  2021-12-13
    • 网络出版时间:  2022-08-24 03:12:01
  • 作者简介:

    赵波(1972—),男,教授,博士. 研究方向:矿山特殊工况无人驾驶感知和定位. E-mail:zhaobo@scu.edu.cn

  • 中图分类号: TP391.4

A Night Image Dehazing Algorithm Based on the Adaptive Atmospheric Light and the Weighted Guided Filter

  • 摘要: 夜间图像去雾对于夜间场景下无人驾驶、交通安防等有重要的工程应用价值。针对暗通道先验算法在夜间雾天场景下失效的问题,提出一种基于自适应大气光和加权引导滤波的夜间图像去雾算法。该算法首先基于图像亮度和饱和度联合求取信道图,并将信道图作为引导图对原图像进行引导滤波得到大气光分布图;为解决暗通道先验在图像亮区域的失效问题,引入亮通道先验矫正亮区域的透射率;为优化亮、暗通道透射率的融合,建立一种基于分段伽马矫正的融合权值计算方法,用于亮区域透射率的权值计算,并利用该透射权值加权得到图像的初始透射率。然后,利用加权聚合引导滤波代替引导滤波细化初始透射率,解决引导滤波弱化细小纹理而引起的边缘模糊问题。最后,将复原图像转换到HSV(Hue–Saturation–Value)色彩空间,对亮度分量V(Value)进行均衡化调整,并对均衡化前后的图像进行线性加权获得最终复原结果。实验结果表明:所提算法大气光分布图估值合理,可有效反映夜间多光源场景下的大气光分布情况;图像亮、暗区域透射率计算准确,复原图像去雾彻底、纹理清晰;与经典算法对比显示,复原结果的峰值信噪比、信息熵、平均梯度和方差的最大提升幅度分别为49.4%、18.3%、172.3%、115.0%,综合指标优于所对比的其他算法。

     

    Abstract: Nighttime image dehazing provides significant industrial contributions to unmanned driving and traffic security. To address the inability of the dark channel prior algorithm facing the foggy night scenes, a night image dehazing algorithm is proposed based on adaptive atmospheric light and weighted guided filtering. The distribution map of the atmospheric light is estimated using the guided filter based on the channel map obtained by image brightness and saturation. Bright channel prior is introduced to correct the transmittance of the bright region to solve the problem of dark channel failure in the bright region. To enhance the fusion of the light and dark channels, a weighted fusion method is used to calculate the weight of the bright region based on piecewise gamma correction, which is further applied to infer the initial transmittance of the raw image. A weighted aggregation guided filtering is proposed to refine the initial transmittance, focusing on addressing the edge blurring caused by the weakened tiny texture of the guided filtering. After converting the intermediate restored image into the HSV domain, image equalization is performed on the V (Value) component to further enhance the image. Finally, the linear weighting algorithm is applied to fuse the enhanced image and the intermediate restored image, to obtain the final recovered image. Experimental results demonstrate that the proposed algorithm is able to properly estimate the atmospheric light distribution map, which can effectively reflect the atmospheric light distribution for the multi-light source in a nighttime scene. The transmittance of both the bright and dark image regions can be accurately calculated to completely recover the raw image with clear texture. In addition, the proposed approach outperforms other comparative baselines, harvesting 49.4%, 18.3%, 172.3%, and 115.0% relative performance improvement on the peak signal-to-noise ratio, information entropy, average gradient, and variance of the restoration results, respectively.

     

  • 户外视频图像采集广泛用于交通安防和无人驾驶等多个领域,但由于雾霾等天气因素的影响,所采集的图像往往会出现模糊、色彩偏移等问题[1]。近年来,针对昼间图像去雾的研究得到了广泛的发展并趋于成熟,但在夜间人工光源条件下,昼间去雾算法存在不能有效复原夜间雾天图像等局限性[2]。以无人驾驶为例,若在夜间有雾场景下不能实现图像的快速去雾,将会严重影响视觉感知模块的性能,从而危及其行驶安全[3]。因此,研究夜间雾天图像复原,弥补现有去雾算法在夜间场景下的不足,对于算法的场景普适性和可靠性具有重要学术意义,同时对于夜间条件无人驾驶和视频安防等设备的正常工作具有重要的工程应用价值。

    目前,图像去雾算法研究主要分为3类,即图像增强、深度学习和图像复原技术。基于图像增强的研究主要基于图像自身信息,通过局部直方图均衡化算法[4]、Retinex算法[5]等提升图像的对比度和色彩纹理信息,但忽略了雾天图像的形成机理,去雾效果有限。程鸿等[6]针对夜间有雾图像光照不均匀问题,利用同态滤波校正图像光照,并使用Retinex算法实现了光源附近的光晕去除,得到了视觉效果较好的夜间无雾图像。得益于计算机计算能力的提升,基于深度学习[7-11]的图像去雾算法不断被衍生出来,但由于不同场景数据库的相对缺乏,限制了算法的普适性和可靠性。Liao等[12]提出一种夜间雾浓度预测网络HDP–Net,并通过预测结果与原图像相减,得到色彩清晰的无雾图像。针对夜间多光源场景,Kuanar等[13]引入了一个基于深度学习的DeGlow–DeHaze迭代模型,该模型同时考虑了不同的颜色和光源,有效地解决了复原图像中的光晕现象。图像复原技术主要利用大气光学模型,在各种先验假设[14-16]基础上,基于雾霾的形成机理逆向推导无雾图像。然而,对于夜间雾天图像,由于多个光源的存在,往往会引起图像照度不均、色彩偏差严重、大气光分布不均匀等问题。因此,在夜间雾天场景下,基于恒定大气光的假设先验有一定的局限性。部分学者尝试将图像增强技术和图像复原技术相融合来实现复原图像的优化[17-19],但算法复杂度也随之增大。Ancuti等[20]提出一种基于多尺度融合的夜间去雾算法,通过在局部图像块上估计图像大气光,避免了光照不均的影响,但图像复原后会引起色彩失真。Li等[21]引入一种新的夜间雾霾模型,通过分析不同光源及其光晕,并在去雾前消除光晕的影响,但该模型在复原后景深处会引起边缘伪影。Yu等[22]首先利用通道差分图作为引导图,对原图像进行引导滤波以得到环境大气光的分布图,并提出亮通道先验矫正亮区域的透射率,最后采用加权融合的方法得到透射率图,但亮、暗通道透射率的不准确融合会引起复原图像中暗区域的纹理消失。Zhang等[23]基于日间室外图像统计,提出最大反射率先验估计环境大气光。随后,Zhang等[24]在此基础上,提出基于最佳尺寸最大反射率先验,有效解决了最大反射率先验在单色区域的失效问题,不足之处是复原后的图像偏暗。本文受Yu等[22]提出的亮、暗通道先验融合思想的启发,提出一种自适应大气光求取方法,主要思想是:基于饱和度和亮度联合求取大气光,同时对亮、暗通道透射率融合方法进行优化;在透射率融合阶段,对亮度分量进行分段伽马矫正得到亮通道透射率融合权值,实现亮、暗通道投射率的可靠融合,并采用加权聚合引导滤波细化初始透射率;为解决暗通道先验引起的复原图像偏暗问题,将复原图像转化到HSV(Hue–Saturation–Value)色彩空间,对亮度分量V(Value)进行均衡化;最后,对均衡化前后的图像进行线性融合,得到最终的复原图像。实验结果表明,所提算法在有效复原图像的同时,去雾彻底,色彩自然,纹理清晰。

    图1为基于自适应大气光和加权聚合引导滤波的夜间图像去雾算法的整体架构,共分为3部分。1)大气光模块:基于亮度和饱和度分量联合求取信道图,并将信道图作为引导图,采用引导滤波对原始图像进行引导得到大气光分布图。2)透射率模块:基于亮通道先验和暗通道先验求取对应的透射率,对亮、暗通道透射率进行线性加权以得到初始透射率,其中,亮通道透射率权值通过对亮度分量进行分段伽马矫正得出;为解决引导滤波容易引起弱纹理模糊等问题,引入加权聚合引导滤波,在引导滤波的聚合阶段基于相似度为不同的窗口分配权值,实现初始透射率的细化。3)亮度矫正模块:为解决复原后图像偏暗的问题,将复原后的图像 ${\boldsymbol{J}}_1 $ 转换到HSV空间,对V分量进行均衡化后得到 ${\boldsymbol{J}}_2 $ ,最后将 ${\boldsymbol{J}}_1 $ ${\boldsymbol{J}}_2 $ 进行线性聚合,得到最终的复原图像 ${\boldsymbol{J}} $

    图  1  基于自适应大气光和加权聚合引导滤波的夜间图像去雾算法框架
    Fig.  1  Framework of night image dehazing algorithm based on the adaptive atmospheric light and the weighted aggregation guided filter
    下载: 全尺寸图片

    Narasimhan等[25]提出的大气散射简化模型广泛应用于图像去雾领域,其物理模型为:

    $$ {\boldsymbol{I}} = {\boldsymbol{J}}{\boldsymbol{t}}+ A({\boldsymbol{O}} - {\boldsymbol{t}})$$ (1)

    式中,I为雾天原始图像矩阵,J为去雾后的图像矩阵,t为透射率矩阵,A为大气光值,O为元素都为1的矩阵。由式(1)可知,At未知,求解J是一个病态问题,对于式(1)的求解需要强有力的先验。但由于在夜间图像场景下,光线主要由人工光源提供,局部大气光分布不均匀,恒定的大气光值不能有效满足去雾需求,Yu等[22]提出一种适用于夜间去雾的大气光学模型,采用大气光分布图A代替恒定大气光值A,即:

    $$ {\boldsymbol{I}} = {\boldsymbol{J}}{\boldsymbol{t}} + {\boldsymbol{A}}({\boldsymbol{O}}- {\boldsymbol{t}}) $$ (2)

    为了解决夜间场景下大气光求解问题,本文提出一种自适应大气光分布图求取方法。首先,联合图像亮度分量和饱和度分量求取大气光信道图L,如式(3)所示;然后,基于引导滤波求取夜间雾天图像的大气光分布图A,如式(4)所示。

    $$ {\boldsymbol{L}}= {\boldsymbol{V}}{\boldsymbol{S}} $$ (3)
    $$ {\boldsymbol{A}} = {\text{GuidFilter}}({\boldsymbol{L}},{\boldsymbol{I}}) $$ (4)

    式中,V为图像亮度分量,S为图像饱和度分量,GuildFilter(L, I)为LI 的引导下进行滤波操作。

    图2中的两行分别为两张夜间有雾图像的大气光分布图。由图2(b)(c)可知,本文所提算法求得的大气光分布结果与Yu等[22]算法一致,说明所提大气光分布图求取算法可以有效求取图像的大气光分布情况,具有一定的鲁棒性。

    图  2  大气光对比
    Fig.  2  Comparison of atmospheric light
    下载: 全尺寸图片

    He等[15]提出的暗通道先验是解决式(1)病态问题最具代表性的假设和先验。该先验表明,在一幅彩色图像的R、G、B 3个通道中,总有1个通道或其邻域内的像素色彩通道强度值很小,近似为0,数学表达式如下:

    $$ {{\boldsymbol{J}}_{{\text{dark}}}} ={\rm{min}}\_D({\boldsymbol{I}}) $$ (5)

    式中:Jdark为像素暗通道矩阵; ${\rm{min}}\_D(\boldsymbol{I})$ 为求解矩阵I的像素暗通道函数,对矩阵I中每一个像素,该函数首先分别计算其邻域像素的R、G、B通道取值的最小值作为每一个邻域像素通道暗值,再计算该像素所有邻域像素通道暗值的最小值,作为该像素的像素暗通道值;本文邻域大小取为15×15。对式(2)两边求取局部最小值,并将式(5)代入式(2),得到暗通道透射率tdcp的表达式:

    $$ {{\boldsymbol{t}}_{{\text{dcp}}}} = 1 - {\rm{min}}\_D \left(\frac{{\boldsymbol{I}}}{\boldsymbol{A}}\right) $$ (6)

    为避免恢复出的图像不协调,引入透射率修正参数ω[15]ω = 0.95,如式(7)所示:

    $$ {{\boldsymbol{t}}_{{\text{dcp}}}} = 1 - {\rm{min}}\_D \left(\frac{{{{\boldsymbol{I}}}}}{{{{\boldsymbol{A}}}}}\right) $$ (7)

    图3为所提算法对夜间雾天图像的复原过程。由图3(a)(b)可知,红框区域存在着多个光源,此处的透射率应相对较大,而暗通道先验求取的透射率却很小,表明暗通道先验在亮区域处出现失效。

    图  3  基于亮、暗通道先验的透射率求取图
    Fig.  3  Transmittance calculation diagrams based on light and dark channel priors
    下载: 全尺寸图片

    为解决夜间亮区域暗通道失效问题,Yu等[22]对500幅夜间有雾图像进行统计发现,在夜间光源等亮区域,95%以上的像素点的亮通道值为0.9 ~ 1.0,即亮通道先验。该先验表明在夜间有雾图像亮区域处,R、G、B 3个通道中总有一个的像素强度值很高,如下所示:

    $$ {{\boldsymbol{J}}_{{\text{bcp}}}} = {\rm{max}}\_B({\boldsymbol{I}}) $$ (8)

    式中, $ {{\boldsymbol{J}}_{{\text{bcp}}}} $ 为图像亮通道矩阵, ${\rm{max}}\_B(\boldsymbol{I})$ 为求解矩阵I的像素亮通道函数。将式(8)代入式(2)得到图像的亮通道透射率tbcp

    $$ {{\boldsymbol{t}}_{{\text{bcp}}}} = {\rm{max}}\_B \left( \frac{{{{\boldsymbol{I}}} - {{\boldsymbol{A}}}}}{{{\boldsymbol{O}} - {{\boldsymbol{A}}}}}\right) $$ (9)

    亮通道透射率如图3(c)所示,亮通道先验在光源等亮区域的透射率估计准确,如红框区域;同时观察图3(c)蓝框中的天空区域,发现亮通道透射率还适用于夜间天空等微光区域的透射率矫正。

    由以上分析可知,亮通道先验更加适用图像中的亮区域;而在非亮区域,暗通道先验更加适用。为同时兼顾两者优点,采用线性加权的方法对亮、暗通道透射率加以融合:

    $$ {\boldsymbol{t}} = {{\boldsymbol{t}}_{{\text{bcp}}}}{\boldsymbol{\mu }} + {{\boldsymbol{t}}_{{\text{dcp}}}}({\boldsymbol{O}} - {\boldsymbol{\mu }}) $$ (10)

    式中, $\;{\boldsymbol{\mu }}$ 为亮通道透射率权值矩阵。为了解决Yu等[22]在亮、暗通道透射率融合不准确引起复原图像中暗区域纹理消失的问题,本文提出一种基于亮度分量的分段伽马矫正权值计算方法,使用图像的亮度值表征图像的亮通道透射率权值。图像亮度图V与图像R、G、B 3通道的像素值的关系如式(11)所示:

    $$ {\boldsymbol{V}} = {\rm{max}}\_C ({\boldsymbol{I}}) $$ (11)

    式中,max_C(I)为求图像I的R、G、B 3通道最大值操作。在夜间场景下,图像成像的主要光源来自人工光源或月光。对于图像中光源处或光源周边的亮区域,其3通道值一般较高;对于图像中距离光源较远的暗区域,其成像的光主要来自月光或微弱的人工光线,其3通道的值一般较低。当图像中某区域的亮度值较大时,3通道值也就越大,说明该区域受光源的影响越大,亮区域的概率就越大,亮通道透射率权值也就越大;反之,暗通道先验应越占优,亮通道透射率也就越小。综合上述分析可得到亮通道透射率权值矩阵 $ \;{\boldsymbol{\mu }} $ 表达式为:

    $$ \left\{ \begin{array}{l} {{\mu}}_{lj}={{{v}}_{lj}}^{0.11}, {{v}}_{lj}\ge T;\\ {{\mu}}_{lj}={{{v}}_{lj}}^{1.1}, {{v}}_{lj}< T\end{array} \right.$$ (12)

    式中, ${{v}}_{lj}$ V的元素,lj为矩阵元素下标,T为亮、暗分割阈值。当图像被分割成亮区域时,其亮区域的透射率权值应越高。为了准确地确定分割阈值,实现图像中亮暗区域的准确分割,选取不同阈值T对图像亮暗区域进行分割并进行权值计算,分割后的亮区域透射率权值图如图4所示。综合分析可知,随着T增大,图4中亮区域的占比逐渐减小。当T为0.3时,图4中的部分非亮区域被分割为亮区域,如图4(b)第2行中的红色区域所示。当T为0.5时,图4(d)中的亮区域被误分割为了暗区域,如图4(d)第1行中蓝色区域所示。当T为0.6时,误分割的情况进一步加重。当T为0.4时,图像中亮区域分割准确,可实现图像亮、暗区域的有效辨识。因此本文取亮、暗区域分割阈值为0.4。

    图  4  不同亮区域分割阈值下的亮区域透射率权值图
    Fig.  4  Weight diagram of bright region transmittance under different bright region segmentation threshold
    下载: 全尺寸图片

    图3(a)的亮区域透射率权值分布情况如图3(d)所示,融合后得到的初始透射率如图3(e)所示。由图3(e)可知,所提权值计算方法对亮区域响应敏感,融合后的透射率在亮、暗区域分布准确。

    为细化透射率,保持复原图像的纹理信息,He等[15]提出引导滤波。但引导滤波在聚合阶段不能考虑全局信息,会导致细小纹理被弱化,出现Halo效应[26-27]。为清晰地保留图像边缘的弱纹理信息,陈彬等[27]在引导滤波的聚合阶段为不同的窗口分配权值,邻域像素k与中心像素i相似度越高,权值越大,像素i的加权聚合引导滤波值qi的数学表达式为:

    $${\qquad {{{q}}_{{i}}} = {{{g}}_{{i}}}\sum\limits_{{{k}} \in {\omega _{{i}}}} {\frac{{{\beta _{{k}}}}}{{||{\beta _{{k}}}||}}} {a_{{k}}} + \sum\limits_{{{k}} \in {\omega _{{i}}}} {\frac{{{\beta _{{k}}}}}{{||{\beta _{{k}}}||}}} {b_{{k}}}} $$ (13)

    式中: $q_i $ 为加权聚合引导滤波输出结果矩阵 $\boldsymbol{Q} $ 中像素 i的取值; $g_i$ 为引导图 $\boldsymbol{G}$ 中像素i的取值, $\boldsymbol{G}$ 为原始图像 $\boldsymbol{I} $ 的灰度图矩阵; $\omega_i$ 为以像素i为中心的邻域,i为中心像素;k $\omega_i$ 的邻域像素, $k \in \omega_i $ $ a_k $ $b_k $ 为邻域 $w_i $ 内像素k的线性系数; $ ||{\beta _k}|| $ 为权值归一化系数, $||{{\boldsymbol{\beta}} _k}|| = \displaystyle\sum\limits_{{{k}} \in {\omega _{{i}}}} {{\beta _{{k}}}}$ βk为像素k局部窗口ωk的聚合权值,其表达式为:

    $$ {\beta _{\boldsymbol{k}}} = \exp \left( - \frac{{{e_{\boldsymbol{k}}}}}{\eta }\right) $$ (14)
    $$ {\qquad {e_{{k}}} = \frac{1}{{|\omega_k |}}\sum\limits_{{{k}} \in {\omega _{{k}}}} {{{({a_{{k}}}{{g}_{n}} + {b_{{k}}} - {{{p}}_{{n}}})}^2}} }$$ (15)

    式(14)~(15)中: $\omega_k$ 为以像素k为中心的邻域局部窗口,k为中心像素;n为邻域局部窗口 $\omega_k$ 的邻域像素, $n \in \omega_k $ $\left|\omega_k\right| $ 为局部窗口 $\omega_k $ 内邻域像素的个数; $p_n $ 为输入的透射率图 $\boldsymbol{t} $ 中像素n的取值; $e_k$ $\omega_k $ 中像素n的灰度值 $g_n $ 与透射率 $p_n $ 的均方误差; $ \eta $ 为一个很小的常数,本文取0.03。基于加权聚合引导滤波细化的透射率图如图3(f)所示。

    将大气光分布图A和加权聚合引导滤波细化后的透射率t代入式(2),透射率下限值t0设定为0.1,如下所示:

    $$ {{\boldsymbol{J}}_1} = \frac{{{\boldsymbol{I}} - {\boldsymbol{A}}}}{{{\rm{max}}\_t ({\boldsymbol{t}},{t_0})}} + {\boldsymbol{A}} $$ (16)

    式中,max_t $\left({\boldsymbol{t}}, t_0\right)$ 为求解t中元素取值和 $t_0 $ 的较大值的函数。当 $\boldsymbol{t} $ 中元素大于 $t_0 $ 时,元素值不变;当 $\boldsymbol{t} $ 中元素小于 $t_0 $ 时,用 $t_0 $ 替换该元素。

    复原图像J1图3(g)所示。由图3(g)可知,基于加权聚合引导滤波细化后的透射率弱纹理保持完整,复原后无Halo效应。

    为解决图像去雾后整体偏暗的问题,将复原图像转化到HSV空间,对亮度进行均衡化,并对矫正前后的图像进行线性聚合,如下所示:

    $$ {\boldsymbol{J}} = \kappa {{\boldsymbol{J}}_1} + (1 - \kappa ){{\boldsymbol{J}}_2} $$ (17)

    式中,J为最终复原的图像,J1为均衡化前的图像,J2为均衡化后的图像, $ \kappa $ 为亮度校正聚合权值。由于引入了亮通道先验,暗通道先验导致的图像偏暗问题已得到了一定的改善。为保证复原结果更接近真实复原效果, $ \kappa $ 应为一个相对较大的值。

    为更加综合评价去雾效果,本文选择峰值信噪比、信息熵、平均梯度、复原图像的方差和Fade指标[28]作为客观评价指标,并通过运行时间评价算法的运行效率。峰值信噪比反映了复原图像和原始有雾图像在相同位置的颜色差异情况,其值越大,图像的失真就越小,图像复原效果越好;信息熵反映图像所含信息量的多少,其值越高,说明图像包含的纹理信息越多;平均梯度反映图像对微小细节反差的表达能力,平均梯度越大,说明复原后的图像纹理越清晰;方差可反映图像的灰度分布程度,方差越大,图像对比度越高,去雾的效果越好;Fade指标反映图像含雾量大小,其值越小,图像去雾越彻底。

    为得到最佳的亮度校正权值系数,本文在0.7 ~ 1.0之间,以0.05为步长选取 $ \kappa $ ,不同 $ \kappa $ 对亮度校正后的复原结果(图3(g))的影响如图5所示。

    图  5  不同聚合权值κ下复原结果的客观指标对比
    Fig.  5  Comparison of objective indicators of recovery results at κ different aggregate weights
    下载: 全尺寸图片

    图5可知:当聚合权值增大时,峰值信噪比增加,而信息熵、平均梯度和方差降低;Fade值先减小后增加,且在 $ \kappa $ =0.85时,Fade有最小值。为了兼顾复原结果的客观指标全局最优,选取亮度校正聚合权值为0.85。由最终的复原结果(图3(h))可知,亮度校正后的图像亮度适中,色彩自然,纹理清晰,去雾彻底。

    仿真实验环境如下:CPU型号为Intel Core i7-8750H CPU@2.20 GHz,操作平台为64位 Windows 10。仿真软件环境为MATLAB R2019a。求取大气光分布A所采用引导滤波的直径为64像素;细化透射率阶段,采用加权聚合引导滤波,滤波直径32像素。

    本文选取了5幅不同夜间场景下的雾天图像,分别记为IMG 1~5,并将本文所提算法与He算法[15]、Li算法[21]、Yu算法[22]、Zhang算法[24]等经典算法对比,结果如图6所示。结果表明:He算法[15]可以去雾图像中大部分雾,但复原后图像整体偏暗,且在光源处产生光晕现象,表明了暗通道先验在夜间场景下去雾的局限性,如图6(b)所示。Li算法[21]可有效去除近景处的雾,且复原后的图像亮度适中,但会在景深的纹理边缘处出现伪影,同时在天空区域出现失真,如图6(c)所示。Yu算法[22]去雾后色彩自然,但会引起暗区域图像纹理的消失,如图6(d)所示。Zhang算法[24]可有效去除近景处的雾,但对远景处的去雾一般,且去雾后会出现一定的色偏,如图6(e)所示。由图6第1、2行可知,本文所提算法去雾彻底,有效地解决了暗通道先验在亮区域的失效问题,去雾后在天空等微光区域色彩无失真。由图6的第3行可知,所提算法能够有效去除图像中远景处的雾,纹理过渡自然,无伪影产生。由图6的4、5行可知,所提算法在多光源的雾天场景下,复原图像无色偏,纹理清晰,具有较强的鲁棒性。

    图  6  复原结果对比
    Fig.  6  Comparison of restoration results
    下载: 全尺寸图片

    各算法对5幅图像复原效果的客观评价结果如图7所示。由图7(a)(e)可知:所提算法的峰值信噪比略低于Zhang算法,优于其他对比算法,说明复原结果相对于原始有雾图像的色彩保真度高,去雾后图像原有特征信息保持较好,在IMG1中峰值信噪比相对于Li算法有最大提升,为49.4%;所提算法的信息熵整体优于对比算法,说明复原图像具有更丰富的细节,去雾更彻底,在IMG2中信息熵相对于He算法提升幅度最大,为18.3%;对比梯度和方差可知,所提算法复原图像的平均梯度和方差优于He算法、Yu算法和Zhang算法,略低于Li算法,在IMG1中,平均梯度相对于He算法的提升幅度最大,为172.3%;在IMG5中,方差相对于Zhang算法提升幅度最大,为115.0%,说明复原图像灰度分布更均匀,轮廓更清晰;对比Fade指标可知,复原图像的Fade值均处于相对低值,说明含雾量较少,所提算法复原结果去雾彻底。算法运行时间见表1,由于在引导滤波时引入了加权聚合模块,因此执行效率低于He算法、Yu算法和Zhang算法,但显著优于Li算法。

    图  7  客观评价指标对比
    Fig.  7  Comparison of objective evaluation indexes
    下载: 全尺寸图片
    表  1  算法执行时间对比表
    Table  1  Comparison of algorithm execution efficiency
    图像 执行时间/s
    He算法 Li算法 Yu算法 Zhang算法 本文算法
    IMG1 0.403 9.801 0.362 0.210 0.919
    IMG2 0.392 7.138 0.307 0.220 0.787
    IMG3 0.364 13.194 0.324 0.190 0.797
    IMG4 0.502 11.104 0.465 0.301 1.169
    IMG5 0.621 20.549 0.618 0.458 1.601

    综合对比客观指标和运行时间发现,本文算法去雾效果明显,去雾后图像纹理清晰、色彩自然,整体优于对比算法。

    为了更加综合地评价算法在不同雾浓度下的图像复原性能,选用相同场景下的不同雾浓度夜间有雾图像进行复原。由于现实生活中同一场景下不同雾浓度的夜间有雾图像较难采集,因此本文采用合成数据集进行研究。Zhang等[24]通过统计大量的夜间有雾图像,得到了在真实夜间光源下有雾图像的色彩分布规律,同时通过设定不同的雾浓度系数,合成了少雾、中雾和浓雾3种不同雾浓度的夜间有雾图像。本文选取其合成数据集中的夜间有雾图像进行复原,复原效果如图8所示,图8的第1行为相同场景下不同雾浓度的夜间有雾图像,第2行为对应的复原结果。由图8可知,在不同雾浓度下,所提算法去雾彻底,亮暗区域的透射率值计算准确,复原结果纹理清晰、色彩自然。

    图  8  相同场景下不同雾浓度的夜间有雾图像复原结果
    Fig.  8  Restoration results of night fog images with different fog concentrations in the same scene
    下载: 全尺寸图片

    表2为不同雾浓度下复原图像信息熵的提升幅度对照。在不同雾浓度下,所提算法复原的图像信息熵均能得到有效的提升,说明复原后图像纹理细节等信息能增大,可实现夜间相同场景下不同雾浓度有雾图像的有效复原。

    表  2  不同雾浓度下复原图像信息熵提升幅度
    Table  2  Amplitude of the increase in the entropy of the restored image under different haze concentrations
    雾浓度 原图信息熵/bit 复原图信息熵/bit 提升幅度/%
    少雾 5.97 7.01 17.40
    中雾 5.97 7.16 19.93
    浓雾 6.20 7.38 19.03

    为解决暗通道先验在夜间场景下的失效问题,本文提出一种基于自适应大气光和加权引导滤波的夜间图像去雾算法。首先,基于图像亮度和饱和度得到信道图,并采用引导滤波自适应得到大气光分布图;然后,建立一种新颖的亮暗通道透射率融合方法,实现了初始透射率的准确求取;在透射率细化阶段,采用加权聚合引导滤波细化透射率,解决了引导滤波引起的边缘模糊问题;最后,将复原图像进行亮度矫正得到最终结果。实验结果表明,所提算法大气光估值合理,亮区域透射率计算准确,去雾彻底,复原结果无Halo效应,具有较强的可靠性,且能对相同场景下的不同雾浓度图像实现有效复原。与经典算法结果对比表明,提出算法的综合指标优于对比算法,峰值信噪比、信息熵、平均梯度和方差的最大提升幅度分别为49.4%、18.3%、172.3%和115.0%。后续工作将针对如何提高本算法的执行效率,实现夜间场景下的实时去雾展开进一步研究。

  • 图  1   基于自适应大气光和加权聚合引导滤波的夜间图像去雾算法框架

    Fig.  1   Framework of night image dehazing algorithm based on the adaptive atmospheric light and the weighted aggregation guided filter

    下载: 全尺寸图片

    图  2   大气光对比

    Fig.  2   Comparison of atmospheric light

    下载: 全尺寸图片

    图  3   基于亮、暗通道先验的透射率求取图

    Fig.  3   Transmittance calculation diagrams based on light and dark channel priors

    下载: 全尺寸图片

    图  4   不同亮区域分割阈值下的亮区域透射率权值图

    Fig.  4   Weight diagram of bright region transmittance under different bright region segmentation threshold

    下载: 全尺寸图片

    图  5   不同聚合权值κ下复原结果的客观指标对比

    Fig.  5   Comparison of objective indicators of recovery results at κ different aggregate weights

    下载: 全尺寸图片

    图  6   复原结果对比

    Fig.  6   Comparison of restoration results

    下载: 全尺寸图片

    图  7   客观评价指标对比

    Fig.  7   Comparison of objective evaluation indexes

    下载: 全尺寸图片

    图  8   相同场景下不同雾浓度的夜间有雾图像复原结果

    Fig.  8   Restoration results of night fog images with different fog concentrations in the same scene

    下载: 全尺寸图片

    表  1   算法执行时间对比表

    Table  1   Comparison of algorithm execution efficiency

    图像 执行时间/s
    He算法 Li算法 Yu算法 Zhang算法 本文算法
    IMG1 0.403 9.801 0.362 0.210 0.919
    IMG2 0.392 7.138 0.307 0.220 0.787
    IMG3 0.364 13.194 0.324 0.190 0.797
    IMG4 0.502 11.104 0.465 0.301 1.169
    IMG5 0.621 20.549 0.618 0.458 1.601

    表  2   不同雾浓度下复原图像信息熵提升幅度

    Table  2   Amplitude of the increase in the entropy of the restored image under different haze concentrations

    雾浓度 原图信息熵/bit 复原图信息熵/bit 提升幅度/%
    少雾 5.97 7.01 17.40
    中雾 5.97 7.16 19.93
    浓雾 6.20 7.38 19.03
  • [1] Narasimhan S G,Nayar S K.Vision and the atmosphere[J].International Journal of Computer Vision,2002,48(3):233–254. doi: 10.1023/a:1016328200723
    [2] Liu Yun,Wang Anzhi,Zhou Hao,et al.Single nighttime image dehazing based on image decomposition[J].Signal Processing,2021,183:107986. doi: 10.1016/j.sigpro.2021.107986
    [3] 沈琛,曹风云,杨雪洁.夜间图像去雾研究综述与展望[J].电子测量与仪器学报,2020,34(11):101–114. doi: 10.13382/j.jemi.B2002994

    Shen Chen,Cao Fengyun,Yang Xuejie.Review and prospect of nighttime haze removal[J].Journal of Electronic Measurement and Instrumentation,2020,34(11):101–114 doi: 10.13382/j.jemi.B2002994
    [4] 蒋华伟,杨震,张鑫,等.图像去雾算法研究进展[J].吉林大学学报(工学版),2021,51(4):1169–1181. doi: 10.13229/j.cnki.jdxbgxb20200382

    Jiang Huawei,Yang Zhen,Zhang Xin,et al.Research progress of image dehazing algorithms[J].Journal of Jilin University(Engineering and Technology Edition),2021,51(4):1169–1181 doi: 10.13229/j.cnki.jdxbgxb20200382
    [5] Yao Liping,Pan Zhongliang.The Retinex-based image dehazing using a particle swarm optimization method[J].Multimedia Tools and Applications,2021,80(3):3425–3442. doi: 10.1007/s11042-020-09812-7
    [6] 程鸿,龙伟,李炎炎.基于非物理模型的夜间图像去雾[J].计算机应用与软件,2020,37(12):204–208. doi: 10.3969/j.issn.1000-386x.2020.12.032

    Cheng Hong,Long Wei,Li Yanyan.Nighttime image dehazing based on non-physical model[J].Computer Applications and Software,2020,37(12):204–208 doi: 10.3969/j.issn.1000-386x.2020.12.032
    [7] Cai Bolun,Xu Xiangmin,Jia Kui,et al.DehazeNet:An end-to-end system for single image haze removal[J].IEEE Transactions on Image Processing:A Publication of the IEEE Signal Processing Society,2016,25(11):5187–5198. doi: 10.1109/TIP.2016.2598681
    [8] Li Shaoyi,Lin Jian,Yang Xi,et al.BPFD–Net:Enhanced dehazing model based on Pix2pix framework for single image[J].Machine Vision and Applications,2021,32(6):1–13. doi: 10.1007/s00138-021-01248-9
    [9] Bhola A,Sharma T,Verma N K.DCNet:Dark channel network for single-image dehazing[J].Machine Vision and Applications,2021,32(3):1–11. doi: 10.1007/s00138-021-01173-x
    [10] Wang Keping,Duan Yumeng,Yang Yi.Single image dehazing algorithm based on pyramid mutil-scale transposed convolutional network[J].Systems Science & Control Engineering,2021,9:150–160. doi: 10.1080/21642583.2020.1833780
    [11] Kan Shichao,Zhang Yue,Zhang Fanghui,et al.A GAN-based input-size flexibility model for single image dehazing[J].Signal Processing:Image Communication,2022,102:116599. doi: 10.1016/j.image.2021.116599
    [12] Liao Yinghong,Su Zhuo,Liang Xiangguo,et al.HDP-net:Haze density prediction network for nighttime dehazing[C]//Advances in Multimedia Information Processing–PCM 2018.Hefei:Springer International Publishing,2018:468–480.
    [13] Kuanar S,Rao K,Mahapatra D,et al.Night time haze and glow removal using deep dilated convolutional network[EB/OL](2019–02–03)[2021–11–20].https://doi.org/10.48550/arXiv.1902.00855,2019
    [14] He Kaiming,Sun Jian,Tang Xiaoou.Single image haze removal using dark channel prior[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2011,33(12):2341–2353. doi: 10.1109/TPAMI.2010.168
    [15] He Kaiming,Sun Jian,Tang Xiaoou.Guided image filtering[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2013,35(6):1397–1409. doi: 10.1109/TPAMI.2012.213
    [16] Berman D,Treibitz T,Avidan S.Non-local image dehazing[C]//Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition.Las Vegas:IEEE,2016:1674–1682.
    [17] 陈志恒,严利民,张竞阳.采用自适应全局亮度补偿的夜间去雾算法[J].红外技术,2021,43(10):954–959.

    Chen Zhiheng,Yan Limin,Zhang Jingyang.Nighttime dehazing algorithm with adaptive global brightness compensation[J].Infrared Technology,2021,43(10):954–959
    [18] 张竞阳,严利民,陈志恒.采用暗态点光源模型的夜间去雾算法[J].红外技术,2021,43(8):798–803.

    Zhang Jingyang,Yan Limin,Chen Zhiheng.Nighttime fog removal using the dark point light source model[J].Infrared Technology,2021,43(8):798–803
    [19] 徐兴贵,杨平,刘永利.基于全尺度Retinex算法的夜间图像去雾[J].微电子学与计算机,2017,34(7):132–136. doi: 10.19304/j.cnki.issn1000-7180.2017.07.028

    Xu Xinggui,Yang Ping,Liu Yongli.Night image dehazing based on full-scale retinex algorithm[J].Microelectronics & Computer,2017,34(7):132–136 doi: 10.19304/j.cnki.issn1000-7180.2017.07.028
    [20] Ancuti C,Ancuti C O,de Vleeschouwer C,et al.Night-time dehazing by fusion[C]//Proceedings of the 2016 IEEE International Conference on Image Processing.Phoenix:IEEE,2016:2256–2260.
    [21] Li Yu,Tan R T,Brown M S.Nighttime haze removal with glow and multiple light colors[C]//Proceedings of the 2015 IEEE International Conference on Computer Vision.Santiago:IEEE,2015:226–234.
    [22] Yu Teng,Song Kang,Miao Pu,et al.Nighttime single image dehazing via pixel-wise alpha blending[J].IEEE Access,2019,7:114619–114630. doi: 10.1109/ACCESS.2019.2936049
    [23] Zhang Jing,Cao Yang,Fang Shuai,et al.Fast haze removal for nighttime image using maximum reflectance prior[C]//Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition.Honolulu:IEEE,2017:7016–7024.
    [24] Zhang Jing,Cao Yang,Zha Zhengjun,et al.Nighttime dehazing with a synthetic benchmark[C]//Proceedings of the MM’20:Proceedings of the 28th ACM International Conference on Multimedia.Seattle:ACM,2020:2355–2363.
    [25] Narasimhan S G,Nayar S K.Chromatic framework for vision in bad weather[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.Hilton Head:IEEE,2000:598–605.
    [26] 麻文刚,张亚东,郭进,等.基于L1范式优化透射率和饱和度补偿的去雾方法[J].铁道学报,2020,42(9):92–101. doi: 10.3969/j.issn.1001-8360.2020.09.012

    Ma Wengang,Zhang Yadong,Guo Jin,et al.Defogging method based on L1 paradigm optimized transmittance and saturation compensation[J].Journal of the China Railway Society,2020,42(9):92–101 doi: 10.3969/j.issn.1001-8360.2020.09.012
    [27] 陈彬,谭新成.加权聚合图像引导滤波[J].计算机工程与设计,2020,41(7):1993–1997. doi: 10.16208/j.issn1000-7024.2020.07.030

    Chen Bin,Tan Xincheng.Weighted aggregation guided image filtering[J].Computer Engineering and Design,2020,41(7):1993–1997 doi: 10.16208/j.issn1000-7024.2020.07.030
    [28] Banerjee S,Sinha Chaudhuri S.Nighttime image-dehazing:A review and quantitative benchmarking[J].Archives of Computational Methods in Engineering,2021,28(4):2943–2975. doi: 10.1007/s11831-020-09485-3
图(8)  /  表(2)

本文结构

    /

    返回文章
    返回