Dual-path joint correction network for underwater image enhancement

Abstract

Acquired underwater images often suffer from severe quality degradation, such as color shift and detail loss due to suspended particles’ light absorption and scattering. In this paper, we propose a Dual-path Joint Correction Network (DJC-NET) to cope with the above degenerate issues, preserving different unique properties of underwater images in a dual-branch way. The design of the light absorption correction branch is to improve the selective absorption of light in water and remove color distortion, while the light scattering correction branch aims to improve the blur caused by scattering. Concretely, in the light absorption correction path, we design the triplet color feature extraction module, which balances the triplet color distribution of the degraded image through independent feature learning between R, G, and B channels. In the light scattering correction path, we develop a dual dimensional attention mechanism to extract the texture information from the features, aiming to recover sufficient details by more effective feature extraction. Furthermore, our method utilizes the multi-scale U-net to adaptively fusion features from different paths to generate enhanced images. Extensive visual and objective experimental results demonstrate that our method outperforms state-of-the-art methods in various underwater scenes.