• Laser & Optoelectronics Progress
  • Vol. 58, Issue 16, 1610006 (2021)
Yue Wang, Dexing Wang*, Hongchun Yuan**, Ruoyou Wu, and Peng Gong
Author Affiliations
  • School of Information, Shanghai Ocean University, Shanghai 201306, China
  • show less
    DOI: 10.3788/LOP202158.1610006 Cite this Article Set citation alerts
    Yue Wang, Dexing Wang, Hongchun Yuan, Ruoyou Wu, Peng Gong. Underwater Image Enhancement Based on Pyramid Attention Mechanism and Generative Adversarial Network[J]. Laser & Optoelectronics Progress, 2021, 58(16): 1610006 Copy Citation Text show less

    Abstract

    This study proposes an underwater image enhancement algorithm based on the pyramid attention mechanism and generative adversarial network (GAN) to improve the enhancement effect of underwater images. It uses the generative adversarial network as the basic architecture, and the generative network adopts the encoding and decoding structures and introduces the feature pyramid attention module. The combination of multi-scale pyramid features and attention mechanism can capture richer advanced features to improve model performance, and the structure of the discriminant network is similar to the Markov discriminator. In addition, a multi-loss function including global similarity, content perception, and color perception is constructed to keep the structure, content,and color of the enhanced image consistent with those of the reference image. The experimental results show that the sharpness, color correction,and contrast of underwater images enhanced by the proposed algorithm are improved. The average values of the structural similarity, underwater image quality measurement, and information entropy are 0.7418, 2.9457, and 4.6925, respectively. For subjective perception and objective evaluation indicators, the experimental results of the proposed algorithm are better than that of the comparison algorithm.
    Yue Wang, Dexing Wang, Hongchun Yuan, Ruoyou Wu, Peng Gong. Underwater Image Enhancement Based on Pyramid Attention Mechanism and Generative Adversarial Network[J]. Laser & Optoelectronics Progress, 2021, 58(16): 1610006
    Download Citation