Adaptive Learning Attention Network for Underwater Image Enhancement

Underwater images suffer from color casts and low illumination due to the scattering and absorption of light as it propagates in water. These problems can interfere with underwater vision tasks, such as recognition and detection. We propose an adaptive learning attention network for underwater image enhancement based on supervised learning, named LANet, to solve these degradation issues. First, a multiscale fusion module is proposed to combine different spatial information. Second, we design a novel parallel attention module(PAM) to focus on the illuminated features and more significant color information coupled with the pixel and channel attention. Then, an adaptive learning module(ALM) can retain the shallow information and adaptively learn important feature information. Further, we utilize a multinomial loss function that is formed by mean absolute error and perceptual loss. Finally, we introduce an asynchronous training mode to promote the network’s performance of multinomial loss function. Qualitative analysis and quantitative evaluations show the excellent performance of our method on different underwater datasets. The code is available at: https://github.com/LiuShiBen/LANet.