Dual-branch feature fusion dehazing network via multispectral channel attention

Huachun Jian,Yongjun Zhang,Weihao Gao,Bufan Wang,Guomei Wang
DOI: https://doi.org/10.1007/s13042-023-02055-6
2024-01-06
International Journal of Machine Learning and Cybernetics
Abstract:Image information degradation significantly complicates single-image dehazing. Despite the noteworthy progress of learning-based methods in removing haze, many existing algorithms focus solely on haze removal while neglecting the crucial aspect of image texture recovery, often resulting in the loss of image details and texture blurring. To tackle this issue, we propose a dual-branch feature fusion dehazing network via multi-spectral channel attention (MS-DBFFN). First, we design a multi-spectral image pre-processing module (MS-PPM), maximizing the utilization of correlation features in hazy images by introducing additional frequency components. This process results in preliminary dehazing images. Second, a dual-branch feature fusion module (DBFF) adaptively fuses features from both the hazy image and preliminary dehazing image via an iterative update strategy. During this gradual iterative updating process, the structural features of the preliminary dehazing image become richer as more feature information is assimilated. Third, our backbone network facilitates the sharing of valuable information across multiple scales from the feature map generated by the DBFF module. Finally, an improved contrastive regularization loss (ICR) generates sharper images with enhanced details and textures. Extensive experiments on both real and synthetic datasets demonstrate that the proposed method, MS-DBFFN, achieves state-of-the-art performance on several image dehazing benchmarks.
computer science, artificial intelligence
What problem does this paper attempt to address?