To fully exploit the dependency of information captured from transmission map estimation and image dehazing, this paper proposes a dual vision attention network for jointly estimating transmission map and clear image from hazy image. The network consists of an image dehazing layer and a transmission estimation layer, with each layer containing a recurrent attentive network and an encoder-decoder network. The haze attention map is generated in the image dehazing layer under the supervision of the transmission map in the recurrent attentive network and guides the subsequent encoder-decoder network to estimate the dehazing result. Similarly, with clear image supervision, the recurrent attentive network in the transmission estimation layer generates the scene attention map and guides the subsequent encoder-decoder network to predict the transmission map. Based on this architecture, we adopted the recurrent units in the two recurrent attentive networks to exchange hidden information. The haze concentration information can then be used to estimate the scene attention map, and the scene information can be used to predict the transmission map. The experimental results demonstrate that our method not only achieves a good dehazing effect on both synthetic and real images, but also outperforms the existing methods in terms of quality and quantity. The average processing time for a single hazy image is 0.043 s. Therefore, our method can be used in the engineering practice of image dehazing.
HAN H N , QIAN F , LÜ J W , et al . Aerial image dehazing using improved dark channel prior [J]. Opt. Precision Eng. , 2020 , 28 ( 6 ): 1387 - 1394 . (in Chinese)
TAN W , CAO SH X , QI W W , et al . A haze removal method for high-resolution remote sensing images [J]. Acta Agronomica Sinica , 2019 , 39 ( 3 ): 0301005 . (in Chinese)
ZHANG ZH , LI Q , XU ZH H , et al . Color-line and dark channel based dehazing for remote sensing images [J]. Opt. Precision Eng. , 2019 , 27 ( 1 ): 181 - 190 . (in Chinese)
WANG Y B , YIN SH B , LÜ ZH W . Under image restoration with adaptive background light estimation and non-local prior [J]. Opt. Precision Eng. , 2019 , 27 ( 2 ): 499 - 509 . (in Chinese)
HE K , SUN J , TANG X . Single image haze removal using dark channel prior [J]. IEEE Transactions on Pattern Analysis and Machine Intelligence , 2010 , 33 ( 12 ): 2341 - 2353 .
BERMAN D , AVIDAN S . Non-local image dehazing [C]. Proceedings of the IEEE conference on Computer Vision and Pattern Recognition , 2016 : 1674 - 1682 .
MENG G , WANG Y , DUAN J , et al . Efficient image dehazing with boundary constraint and contextual regularization [C]. Proceedings of the IEEE International Conference on Computer Vision , 2013 : 617 - 624 .
CHEN Y , GUO H G , AI Y P . Single image dehazing method based on multi-scale convolution neural network [J]. Acta Optica Sinica , 2019 , 39 ( 10 ): 1010001 . (in Chinese)
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 .
LI B , PENG X , WANG Z , et al . Aod-net: All-in-one dehazing network [C]. Proceedings of the IEEE International Conference on Computer Vision , 2017 : 4770 - 4778 .
QU Y , CHEN Y , HUANG J , et al . Enhanced pix2pix dehazing network [C]. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition , 2019 : 8160 - 8168 .
LIU X , MA Y , SHI Z , et al . Griddehazenet: Attention-based multi-scale network for image dehazing [C]. Proceedings of the IEEE International Conference on Computer Vision , 2019 : 7314 - 7323 .
ZHANG H , PATEL V M . Densely connected pyramid dehazing network [C]. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition , 2019 : 3194 - 3203 .
ZHANG H , SINDAGIV , PATEL V M . Joint transmission map estimation and dehazing using deep networks [J]. IEEE Transactions on Circuits and Systems for Video Technology , 2019 .
REN H , EI-KHAMY M , LEE J . Dn-resnet: Efficient deep residual network for image denoising [C]. Asian Conference on Computer Vision . Springer , Cham , 2018 : 215 - 230 .
CHEN Q J , ZHANG X , CHAI Y ZH . Image defogging algorithms based on multiscale convolution neural network [J]. Chinese Journal of Liquid Crystals and Displays , 2019 , 34 ( 2 ): 220 - 227 . (in chinese)
SILBERMAN N , HOIEM D , KOHLI P , et al . Indoor segmentation and support inference from rgbd images [C]. European conference on computer vision . Springer , Berlin, Heidelberg , 2012 : 746 - 760 .
LI B , REN W , FU D , et al . Benchmarking single-image dehazing and beyond [J]. IEEE Transactions on Image Processing , 2018 , 28 ( 1 ): 492 - 505 .