Chinese Journal of Liquid Crystals and Displays, Volume. 38, Issue 3, 378(2023)

Lightweight underwater image enhancement network based on GAN

Hao-xuan LIU1,3, Shan-ling LIN1,3, Zhi-xian LIN1,2,3, Tai-liang GUO2,3, and Jian-pu LIN1,3、*
Author Affiliations
  • 1College of Advanced Manufacturing,Fuzhou University,Quanzhou 362000,China
  • 2College of Physics and Information Engineering,Fuzhou University,Fuzhou 350116,China
  • 3Fujian Science & Technology Innovation Laboratory for Optoelectronic Information of China,Fuzhou 350116,China
  • show less

    Due to the absorption and scattering of underwater light, the underwater image suffers from distortion and loss of details, which seriously affects the detection and recognition of subsequent underwater target. In this paper, a lightweight fully convolutional layer generative adversarial neural network DUnet-GAN is proposed to enhance underwater image. According to the characteristics of underwater image, this paper proposes a multi-task objective function, which enables the model to enhance the image quality by perceiving the overall content, color, local texture and style information of the image. In addition, we compare DUnet-GAN with some important existing models and make a quantitative evaluation. The results show that in EUVP dataset, the PSNR of the proposed model is above 26 dB, the SSIM is 0.8, and the number of parameters is 11 MB, which is only 5% of the number of parameters of other models with the same performance and better than the FunIE-GAN with 26 MB parameters. Meanwhile, UIQM is 2.85, second only to Cycle-GAN model, and the enhancement effect is significant subjectively. More importantly, the enhanced image provides better performance for underwater target detection and other models, and also meets the lightweight requirements of models for equipment such as underwater robots.

    Tools

    Get Citation

    Copy Citation Text

    Hao-xuan LIU, Shan-ling LIN, Zhi-xian LIN, Tai-liang GUO, Jian-pu LIN. Lightweight underwater image enhancement network based on GAN[J]. Chinese Journal of Liquid Crystals and Displays, 2023, 38(3): 378

    Download Citation

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

    Category: Research Articles

    Received: Jun. 24, 2022

    Accepted: --

    Published Online: Apr. 3, 2023

    The Author Email: Jian-pu LIN (ljp@fzu.edu.cn)

    DOI:10.37188/CJLCD.2022-0212

    Topics