17 June 2021 RGNAM: recurrent grid network with an attention mechanism for single-image dehazing
Nian Wang, Zhigao Cui, Yanzhao Su, Aihua Li
Author Affiliations +
Abstract

Single-image dehazing is a critical problem since haze existence degrades the quality of images and hinders most advanced computer vision tasks. Early methods solve this problem via the atmospheric scattering model, which estimate the intermediate parameters and then recover a clear image by low-level priors or learning on synthetic datasets. However, these model-based methods do not hold in various scenes. Recently, many learning-based methods have directly recovered dehazed images from inputs, but these methods fail to deal with dense haze and always lead to color distortion. To solve this problem, we build a recurrent grid network with an attention mechanism, named RGNAM. Specifically, we propose a recurrent feature extraction block, which repeats a local residual structure to enhance feature representation and adopts a spatial attention module to focus on dense haze. To alleviate color distortion, we extract local features (e.g., structures and edges) and global features (e.g., colors and textures) from a grid network and propose a feature fusion module combining trainable weights and channel attention mechanisms to merge these complementary features effectively. We train our model with smooth L1 loss and structural similarity loss. The experimental results demonstrate that our proposed RGNAM surpasses previous state-of-the-art single-image dehazing methods on both synthetic and real haze datasets.

© 2021 SPIE and IS&T 1017-9909/2021/$28.00© 2021 SPIE and IS&T
Nian Wang, Zhigao Cui, Yanzhao Su, and Aihua Li "RGNAM: recurrent grid network with an attention mechanism for single-image dehazing," Journal of Electronic Imaging 30(3), 033026 (17 June 2021). https://doi.org/10.1117/1.JEI.30.3.033026
Received: 20 March 2021; Accepted: 3 June 2021; Published: 17 June 2021
Lens.org Logo
CITATIONS
Cited by 4 scholarly publications.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Air contamination

Atmospheric modeling

Feature extraction

Distortion

Visualization

Scattering

Image fusion

RELATED CONTENT


Back to Top