Pyramid Channel-based Feature Attention Network for image dehazing

• Published in: Computer vision and image understanding Vol. 197-198; p. 103003
• Main Authors: Zhang, Xiaoqin, Wang, Tao, Wang, Jinxin, Tang, Guiying, Zhao, Li
• Format: Journal Article
• Language: English
• Published: Elsevier Inc 01.08.2020
• Subjects: Channel attention; Deep neural network; Image dehazing; Image dehazing; Deep neural network; Channel attention
• ISSN: 1077-3142; 1090-235X
• DOI: 10.1016/j.cviu.2020.103003

Traditional deep learning-based image dehazing methods usually use the high-level features (which contain more semantic information) to remove haze in the input image, while ignoring the low-level features (which contain more detail information). In this paper, a Pyramid Channel-based Feature Attention Network (PCFAN) is proposed for single image dehazing, which leverages complementarity among different level features in a pyramid manner with channel attention mechanism. PCFAN consists of three modules: a three-scale feature extraction module, a pyramid channel-based feature attention module (PCFA), and an image reconstruction module. The three-scale feature extraction module simultaneously captures the low-level spatial structural features and the high-level contextual features in different scales. The PCFA module utilizes the feature pyramid and the channel attention mechanism, which effectively extracts interdependent channel maps and selectively aggregates the more important features in a pyramid manner for image dehazing. The image reconstruction module is used to reconstruct features to recover a clear image. Meanwhile, a loss function that combines a mean square error loss part and an edge loss part is employed in PCFAN, which can better preserve image details. Experimental results demonstrate that the proposed PCFAN outperforms existing state-of-the-art algorithms on standard benchmark datasets in terms of accuracy, efficiency, and visual effect. The code will be made publicly available. •We propose an end-to-end Pyramid Channel-based Feature Attention Network for single image dehazing, which does not need to explicitly estimate the transmission map and the atmospheric light.•The PCFA module can extract more informative features by the channel attention block, and fuse the complementary features in different levels in a pyramid manner.•A loss function that combines a mean square error loss part and an edge loss part is employed in PCFAN, which can better preserve image details.•Extensive experiments demonstrate that the proposed PCFAN performs favorably compared with state-of-the-art methods, in terms of quantitative accuracy and qualitative visual effect.