Multi-scale network with attention mechanism for underwater image enhancement

Abstract

Underwater images suffer from severe degradation due to the complicated environments. Though numerous approaches have been proposed, accurately correcting color bias meanwhile effectively enhancing contrast is still a difficult problem. In order to address this issue, a dedicated designed Multi-scale Network with Attention mechanism (MNA) is introduced in this work. Concretely, MNA contains four key characteristics: (a) setting more convolution layers in shallow flows, (b) letting connections from high-level to adjacent low-level stream progressively, (c) simplifying dual attention mechanism embedding it in conventional residual block, (d) exploiting channel attention module to fuse multi-scale information rather than conventional summation operation. Extensive experiments demonstrate that our MNA achieves better performance than some well-recognized technologies. Meanwhile, ablation study proves the effectiveness of each component in our MNA. In addition, extended applications demonstrate the improvement of our MNA in local feature points matching and image segmentation.