Laser & Optoelectronics Progress, Volume. 62, Issue 14, 1401003(2025)

Underwater Image Enhancement Algorithm Using Color Correction and Attention Mechanism

Ziqian Peng1,2, Zhiyuan Cheng1,3、*, and Jiahao Hu1,2
Author Affiliations
  • 1Photoelectric Tracking and Measurement Technology Laboratory, Xi'an Institute of Optics and Precision Mechanics of Chinese Academy of Sciences, Xi'an 710119, Shaanxi , China
  • 2University of Chinese Academy of Sciences, Beijing 100049, China
  • 3Key Laboratory of Space Precision Measurement Technology, Chinese Academy of Sciences, Xi'an 710119, Shaanxi , China
  • show less

    A generative adversarial network model based on color correction preprocessing and an attention mechanism is proposed to address the challenges of image degradation and color shift in underwater imaging owing to scattering and absorption. In the preprocessing stage, an unsupervised image color enhancement module (UCM) is implemented to correct the color distortion of underwater images. The UCM is introduced to preprocess and enhance underwater image enhancement benchmark (UIEB) dataset, which effectively mitigates the issue of color shift. Then, in the generating adversarial network architecture, channel and spatial dual attention modules are used for feature attention of the network, thereby optimizing the ability of the image generation network for learning underwater image characteristics. Furthermore, a composite loss function is designed in the generative adversarial network to improve the performance of the generator and discriminator. Finally, enhanced images are obtained through training and testing on the underwater dataset. Compared with traditional underwater image enhancement algorithms, the proposed algorithm demonstrates better visual effects in subjective evaluations. Objective indicators show substantial improvements in the UIEB test set, with the peak signal-to-noise ratio, structural similarity index, and underwater image quality index increasing by 12%, 44%, and 12%, respectively, compared with the best-performing referenced algorithm. Experimental results show that the proposed algorithm can effectively correct color deviations in underwater images and improve image quality.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Ziqian Peng, Zhiyuan Cheng, Jiahao Hu. Underwater Image Enhancement Algorithm Using Color Correction and Attention Mechanism[J]. Laser & Optoelectronics Progress, 2025, 62(14): 1401003

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Atmospheric Optics and Oceanic Optics

    Received: Nov. 12, 2024

    Accepted: Feb. 7, 2025

    Published Online: Jul. 16, 2025

    The Author Email: Zhiyuan Cheng (czy@opt.ac.cn)

    DOI:10.3788/LOP242255

    CSTR:32186.14.LOP242255

    Topics