Underwater image enhancement with global–local networks and compressed-histogram equalization

• Published in: Signal processing. Image communication Vol. 86; p. 115892
• Main Authors: Fu, Xueyang, Cao, Xiangyong
• Format: Journal Article
• Language: English
• Published: Amsterdam Elsevier B.V 01.08.2020; Elsevier BV
• Subjects: CNNs; Color; Computer architecture; Deep learning; Electromagnetic absorption; Equalization; Histograms; Image contrast; Image enhancement; Image quality; Machine learning; Underwater; Underwater image; Deep learning; Underwater image; Image enhancement; CNNs
• ISSN: 0923-5965; 1879-2677
• DOI: 10.1016/j.image.2020.115892

Due to the light absorption and scattering, captured underwater images usually contain severe color distortion and contrast reduction. To address the above problems, we combine the merits of deep learning and conventional image enhancement technology to improve the underwater image quality. We first propose a two-branch network to compensate the global distorted color and local reduced contrast, respectively. Adopting this global–local network can greatly ease the learning problem, so that it can be handled by using a lightweight network architecture. To cope with the complex and changeable underwater environment, we then design a compressed-histogram equalization to complement the data-driven deep learning, in which the parameters are fixed after training. The proposed compression strategy is able to generate vivid results without introducing over-enhancement and extra computing burden. Experiments demonstrate that our method significantly outperforms several state-of-the-arts in both qualitative and quantitative qualities. •The proposed method integrates deep learning and handcrafted image enhancement.•The proposed method has a lightweight architecture for underwater image enhancement.•The proposed method improves image contrast without over-enhancement.•The proposed method has a lightweight architecture and efficient computation time.