• Electronics Optics & Control
  • Vol. 30, Issue 11, -1 (2023)
CHEN Haixiu1, LU Kang2, HE Shanshan2, HUANG Zijie2, and FANG Weizhi2
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less
    DOI: 10.3969/j.issn.1671-637x.2023.11.009 Cite this Article
    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 Copy Citation Text show less

    Abstract

    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.
    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