Underwater Image Enhancement Network Based on Dual Layers Regression

Abstract

Due to the absorption and scattering of light in water, captured underwater images often suffer from some degradation, such as color cast, blur, and low contrast. These types of degradation usually affect and degrade the performance of computer vision methods and tasks under water. In order to solve these problems, in this paper, we propose a multi-stage and gradually optimized underwater image enhancement deep network, named DLRNet, based on dual layers regression. Our network emphasizes important information by aggregating different depth features in the channel attention module, and the dual-layer regression module is designed with regression to obtain the ambient light and scene light transmission for an underwater image. Then, with the underwater imaging model, the enhanced underwater image for a degraded image can be obtained with normal color, higher clarity, and contrast. The experiments on some different datasets with qualitative analysis and quantitative evaluations validate our network, and show that it outperforms some state-of-the-art approaches.