Multi-scale Dehazing Network Based on Error-backward Mechanism

被引:0
|
作者
Yang A.-P. [1 ]
Li X.-X. [1 ]
Zhang T.-F. [1 ]
Wang C.-C. [1 ]
Wang J. [1 ]
机构
[1] School of Electrical and Information Engineering, Tianjin University, Tianjin
来源
基金
中国国家自然科学基金;
关键词
deep learning; error-backward; Image dehazing; multi-scale network;
D O I
10.16383/j.aas.c210264
中图分类号
学科分类号
摘要
The existing dehazing methods can not estimate large-scale target features accurately due to the loss of spatial context information, leading to the destruction of image structure and remaining of haze. To solve this problem, we propose a novel multi-scale dehazing network based on error-backward mechanism, which is composed of error-backward multi-scale dehazing group (EMDG), gated fusion module (GFM) and optimization module. Error-backward multi-scale dehazing group consists of error-backward block (EB) and haze aware unit (HAU). With comparing the difference between feature maps of the coarse-scale sub-network and those of the fine-scale sub-network, error-backward block produces an error map and then transmits it to the last coarse-scale sub-network. So the structure and context information can be reused effectively. Haze aware unit is the core of all sub-networks, which consists of residual dense blocks (RDB) and haze density adaptive detection block (HDADB). It helps to extract local information and accomplish adaptive dehazing according to haze density. Differently from the existing fusion-based methods stacking features from different scales directly, the proposed gated fusion module learns the optimal weights of feature maps from different sub-networks, which prevents interferences to destroy image structure and details. The output of optimization module will be the final dehazed image. Extensive experiments on synthetic datasets and real datasets validate the superiority of our proposed network, especially for the haze removal at a distant view. © 2023 Science Press. All rights reserved.
引用
收藏
页码:1857 / 1867
页数:10
相关论文
共 28 条
  • [1] Wonkyun K, Jong Y, Jechang J., Contrast enhancement using histogram equalization based on logarithmic mapping, Optical Engineering, 51, 6, (2012)
  • [2] Li H, Xie W H, Wang X G., GPU implementation of multi-scale retinex image enhancement algorithm, Proceedings of the IEEE/ACS International Conference of Computer Systems and Applications, pp. 1-5, (2016)
  • [3] Stark J A., Adaptive image contrast enhancement using generalizations of histogram equalization, IEEE Transactions on Image Processing, 9, 5, pp. 889-896, (2000)
  • [4] He K, Sun J, Tang X., Single image haze removal using dark channel prior, IEEE Transactions on Pattern Analysis and Machine Intelligence, 33, 12, pp. 2341-2353, (2010)
  • [5] Tarel J P, Hautiere N., Fast visibility restoration from a single color or gray level image, Proceedings of the IEEE International Conference on Computer Vision, pp. 2201-2208, (2010)
  • [6] Zhu Q S, Mai J M, Shao L., A fast single image haze removal algorithm using color attenuation prior, IEEE Transactions on Image Processing, 24, 11, pp. 3522-3533, (2015)
  • [7] Berman D, Treibitz T, Avidan S., Non-local image dehazing, Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 1674-1682, (2016)
  • [8] Zhang Xiao-Gang, Tang Mei-Ling, Chen Hua, Tang Hong-Zhong, A dehazing method in single image based on double-area filter and image fusion, Acta Automatica Sinica, 40, 8, pp. 1733-1739, (2014)
  • [9] Wang Yun-Fei, Feng Guo-Qiang, Liu Hua-Wei, Zhao Bo-Xin, Super-pixel-based mean and mean square deviation dark channel for single image fog removal, Acta Automatica Sinica, 44, 3, pp. 481-489, (2018)
  • [10] Cai B L, Xu X M, Jia K, Qing C M, Tao D C., DehazeNet: An end-to-end system for single image haze removal, IEEE Transactions on Image Processing, 25, 11, pp. 5187-5198, (2016)