Coarse-to-fine Underwater Image Enhancement with Lightweight CNN and Attention-based Refinement

Ali Khandouzi,Mehdi Ezoji
DOI: https://doi.org/10.1016/j.jvcir.2024.104068
IF: 2.887
2024-01-30
Journal of Visual Communication and Image Representation
Abstract:This paper presents a deep learning-based underwater image enhancement method supported by a classical image processing technique. The proposed method includes an end-to-end three-module structure. The first module is a lightweight two-branch network that retrieves lost colors and to some extent overall appearance through the global and local image enhancement. The second module is the modified histogram equalization to improve the global intensity, contrast and local colors of the image by controlling over-intensity and artificial colors that may result from histogram equalization. The last part is the attention module, utilized to help the proposed framework have a synergistic combination of the previous modules. The attention module is designed to combine the advantages of the previous modules and evade their drawbacks. Experiments to objectively and subjectively evaluate the performance of the proposed model show that the proposed model is superior to existing underwater image enhancement methods.
computer science, information systems, software engineering
What problem does this paper attempt to address?