Underwater Image Enhancement Via Multicolor Space-Guided Curve Estimation

Shuyu Hao,Jichang Guo,Guanhua An,Yudong Wang

DOI: https://doi.org/10.1016/j.jvcir.2024.104240

IF: 2.887

2024-01-01

Journal of Visual Communication and Image Representation

Abstract:High-quality images are of prime importance for vision tasks in underwater environment, ocean exploitation and marine eco-environment protection. However, due to attenuation and scattering related to wavelength and distance, underwater images often suffer from severe distortion, like color cast and loss of detail, challenging the capture of high-quality images. Existing methods mostly focus on pixel-wise mapping from input to output images, which disrupts the latent relationships between neighboring pixels and suffers limited flexibility of linear mapping. Additionally, most methods operate solely within RGB color space, which is insensitive to some image properties such as luminance and saturation, may be inconsistent with human visual perception. In this paper, we combine global and local enhancement, formulating underwater image global enhancement as a task of image-specific piecewise curve estimation based on a deep network, introducing three color spaces (RGB, HSV, CIELab). Our network learns piecewise nonlinear curves specific to different channels of multiple color spaces, thereby introducing highly flexible nonlinear mapping and achieving targeted adjustment. Besides, we employ channel-wise attention mechanism to allocate weights for different adjustment results from multiple color spaces, combining their advantages to further enhance image properties such as luminance, saturation, and color, aiming to improve image quality and align it more closely with human visual perception. To assess the performance of the network, extensive experiments are conducted on synthetic and real-world underwater image datasets, with comparisons made against state-of-the-art methods. Both quantitative and qualitative results indicate our network’s remarkable performance in both visual quality and quantitative metrics.