Electronics Optics & Control, Volume. 30, Issue 11, -1(2023)

An Underwater Image Enhancement Algorithm Based on Multi-Scale Triple Attention

CHEN Haixiu1, LU Kang2, HE Shanshan2, HUANG Zijie2, and FANG Weizhi2
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less

    To solve the problems of color distortion and loss of details in underwater images, an underwater image enhancement algorithm based on Multi-Scale Triple Attention (MSTA) is proposed.The algorithm uses the Generative Adversarial Network (GAN) as the basic architecture, and the generative network adopts the encoding and decoding structure.An MSTA module is designed.The combination of the multi-scale structure and the Triple Attention (TA) mechanism can realize the cross-dimensional interaction of information at different levels, making the network better learn the features of underwater images and suppress the features of noise.The discriminant network adopts a structure similar to Markov discriminator.Multiple loss functions are constructed to make the generated image consistent with the reference image in terms of structure, content and color.The experimental results show that the proposed algorithm is superior to the comparison algorithms in terms of subjective visual effects and objective evaluation indicators.The proposed algorithm can effectively improve the feature extraction ability of the network, restore the color of underwater images in different scenes, and enhance the contrast and clarity of the images.

    Tools

    Get Citation

    Copy Citation Text

    CHEN Haixiu, LU Kang, HE Shanshan, HUANG Zijie, FANG Weizhi. An Underwater Image Enhancement Algorithm Based on Multi-Scale Triple Attention[J]. Electronics Optics & Control, 2023, 30(11): -1

    Download Citation

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

    Received: Nov. 22, 2022

    Accepted: --

    Published Online: Jan. 20, 2024

    The Author Email:

    DOI:10.3969/j.issn.1671-637x.2023.11.009

    Topics