该文章提出一种名为DehazeNet的可训练的端到端系统,用于传输值估计。 DehazeNet将模糊图像作为输入,并输出其中间透射图,随后用于通过大气散射模型恢复无雾图像。 DehazeNet采用基于卷积神经网络的深层架构,其层专门设计用于体现图像去雾中已建立的假设/先验。具体而言,Maxout单位的图层用于特征提取,这可以生成几乎所有与雾相关的特征。我们还在DehazeNet中提出了一种新的非线性激活函数,称为双边整流线性单元,它能够提高恢复的无雾图像的质量。我们在提议的DehazeNet的组件与现有方法中使用的组件之间建立连接。基准图像的实验表明,DehazeNet比现有方法具有更高的性能,同时保持高效和易用。

DehazeNet架构图

1 摘要

背景:Single image haze removal is a challenging ill-posed problem.

现存方法:Existing methods use various constraints/priors to get plausible dehazing solutions. The key to achieve haze removal is to estimate a medium transmission map for an input hazy image.

提出的方法:In this paper, we propose a trainable end-to-end system called DehazeNet, for medium transmission estimation. DehazeNet takes a hazy image as input, and outputs its medium transmission map that is subsequently used to recover a haze-free image via atmospheric scattering model. DehazeNet adopts convolutional neural network-based deep architecture, whose layers are specially designed to embody the established assumptions/priors in image dehazing. Specifically, the layers of Maxout units are used for feature extraction, which can generate almost all haze-relevant features. We also propose a novel nonlinear activation function in DehazeNet, called bilateral rectified linear unit, which is able to improve the quality of recovered haze-free image. We establish connections between the components of the proposed DehazeNet and those used in existing methods. Experiments on benchmark images show that DehazeNet achieves superior performance over existing methods, yet keeps efficient and easy to use. 

2 贡献

  • DehazeNet是一个端到端系统。 它直接学习和估计模糊图像patches与其传输图之间的映射关系。 这是通过其深层架构的特殊设计来实现的,以体现已建立的图像去雾原理。
  • 提出了一种新的非线性激活函数,称为双边整流线性单元1(BReLU)。 BReLU扩展了整流线性单元(ReLU)并证明了其在获得精确图像恢复方面的重要性。 从技术上讲,BReLU使用双边约束来减少搜索空间并改善收敛。
  • 在DehazeNet的组件与现有的去雾方法中使用的假设/先验之间建立联系,并解释DehazeNet通过自动从头到尾自动学习所有这些组件来改进这些方法。 

3 图像去雾的核心

现有的图像去雾(Image Dehazing)技术离不开一个简单的自然模型——大气散射模型(Atmospheric Scattering Model)。大气散射模型描述了,在雾霾和光照的共同作用下的成像机制:

大气散射模型

其中,$t(x)$ 是媒介透射率(medium transmission),顾名思义表示能顺利透过雾霾到达摄像头的比率。因此,透射率跟物体与摄像头距离 $d(x)$ 成反比,离摄像头越远的物体受雾霾影响更大。当距离 $d(x)$ 趋于无穷大时,透射率 $t(x)$ 趋于零,$I(x)$ 趋近于 $\alpha$,其中:

综上所述,去雾的核心是如何更精确地估计媒介透射率 t(x)。

4 基于人工特征

手工特征是传统机器视觉的基础,讲究的是熟能生巧,依赖的是实践出真知。通过“观察→经验→设计”构建各式各样的特征来满足各式各样的任务需求。图像去雾技术也是沿着手工特征逐步地发展起来。

(1)暗通道先验[2](Dark Channel Prior,DCP)

说起去雾特征,不得不提起的暗通道先验(DCP)。大道之行在于简,DCP作为CVPR 2009的最佳论文,以简洁有效的先验假设解决了雾霾浓度估计问题。

观察发现,清晰图像块的RGB颜色空间中有一个通道很暗(数值很低甚至接近于零)。因此基于暗通道先验,雾的浓度可由最暗通道的数值近似表示:

(2)最大对比度[3](Maximum Contrast,MC)

根据大气散射模型,雾霾会降低物体成像的对比度:Σx‖ΔI(x)‖=tΣx‖ΔJ(x)‖≤Σx‖ΔJ(x)‖。因此,基于这个推论可利用局部对比度来近似估计雾霾的浓度。同时,也可以通过最大化局部对比度来还原图像的颜色和能见度。

(3)颜色衰减先验[4](Color Attenuation Prior,CAP)

颜色衰减先验(CAP)是一种与暗通道先验(DCP)相似的先验特征。观察发现雾霾会同时导致图像饱和度的降低和亮度的增加,整体上表现为颜色的衰减。根据颜色衰减先验,亮度和饱和度的差值被应用于估计雾霾的浓度:

颜色衰减先验

(4)色度不一致(Hue Disparity)[5]

5 基于深度智能

人的视觉系统并不需依赖这些显式的特征变换,便可以很好地估计雾的浓度和场景的深度。DehazeNet是一个特殊设计的深度卷积网络,利用深度学习去智能地学习雾霾特征,解决手工特征设计的难点和痛点。

DehazeNet架构图

网络的设计

  1. 第一层是特征提取层,即提取有雾图像特征。根据不同的假设与先验设计不同的滤波器。举的例子中有16个滤波器。其中每四个是上述一种先验特征滤波器。通过maxout unit的激活函数,每四个输出一张图。这里不padding,输入是31616三通道的块。输出的是四个161212,每一个代表一种特征。

  2. 使用多尺度的平行卷积操作。由于多尺度特征被证明有利于去雾并且在inception的模型中也用到了平行卷积,即同一张图用不同尺度的卷积核进行卷积。分别用16个33、16个55和16个77的卷积核进行卷积,每一种尺度产生16个,并且通过padding每张图大小应该是一致的。总共获得48个4810*10。

  3. Maxpooling对局部数据敏感,另外根据假设透射率有局部不变性,所以用一个77局部最大值滤波替代maxpooling。输出是48个486*6。

  4. 通过1个44的卷积核,产生11的标量,并且使用的激活函数为BReLU。因为ReLU抑制了小于0的数,只适用于图像分类等方面,并不适合图像复原。因为最后的透射率图允许高于1或者低于0。所以提出了BReLU,既保持了局部线性,又保持了双边的限制。输出的是一个标量,即输入块中心点的透射率值。

BReLU

参考文献

[1] Cai B, Xu X, Jia K, et al. DehazeNet: An End-to-End System for Single Image Haze Removal [J]. IEEE Transactions on Image Processing, 2016, 25(11): 5187-5198.

[2] He K, Sun J, Tang X. Single image haze removal using dark channel prior[J]. IEEE transactions on pattern analysis and machine intelligence, 2011, 33(12): 2341-2353.

[3] Tan R T. Visibility in bad weather from a single image[C]//Computer Vision and Pattern Recognition, 2008. CVPR 2008. IEEE Conference on. IEEE, 2008: 1-8.

[4] Zhu Q, Mai J, Shao L. A fast single image haze removal algorithm using color attenuation prior[J]. IEEE Transactions on Image Processing, 2015, 24(11): 3522-3533.

[5] C. O. Ancuti, C. Ancuti, C. Hermans, and P. Bekaert, “A fast semiinverse approach to detect and remove the haze from a single image,” in Computer Vision–ACCV 2010, 2011, pp. 501–514.

[6] Goodfellow I J, Warde-Farley D, Mirza M, et al. Maxout networks[J]. ICML (3), 2013, 28: 1319-1327.

[7] DehazeNet: An End-to-End System for Single Image Haze Removal解读 https://blog.csdn.net/Julialove102123/article/details/80199276

[8] 阅读笔记—-DehazeNet:An End-to-End System for Single Image Haze Removal https://blog.csdn.net/yinhou1771/article/details/82682311