Laser & Optoelectronics Progress, Volume. 58, Issue 16, 1610006(2021)

Underwater Image Enhancement Based on Pyramid Attention Mechanism and Generative Adversarial Network

Yue Wang, Dexing Wang*, Hongchun Yuan**, Ruoyou Wu, and Peng Gong
Author Affiliations
  • School of Information, Shanghai Ocean University, Shanghai 201306, China
  • show less
    Figures & Tables(11)
    Feature pyramid attention module
    Structure of FPAGAN
    Generative network structure
    Discriminative network structure
    Results of comparative experiments without FPA module and with FPA module. (a) Underwater images; (b) GAN; (c) FPAGAN
    Qualitative comparison of different methods on test set A. (a) Underwater images; (b) GC; (c) UDCP; (d) LDCP; (e) UWCNN; (f) FUnIE-GAN; (g) proposed method; (h) reference
    Qualitative comparison of different methods on test set B. (a) Underwater images; (b) GC; (c) UDCP; (d) LDCP; (e) UWCNN; (f) FUnIE-GAN; (g) proposed method
    • Table 1. Experimental results obtained by GAN and FPAGAN on test set A

      View table

      Table 1. Experimental results obtained by GAN and FPAGAN on test set A

      MethodPSNRSSIM
      GAN21.93280.7330
      FPAGAN22.39840.7418
    • Table 2. Experimental results obtained by GAN and FPAGAN on test set B

      View table

      Table 2. Experimental results obtained by GAN and FPAGAN on test set B

      MethodUIQMIENIQE
      GAN2.90864.688942.5059
      FPAGAN2.94574.692537.6927
    • Table 3. Quantitative comparison of different methods on test set A

      View table

      Table 3. Quantitative comparison of different methods on test set A

      MetricsGCUDCPLDCPUWCNNFUnIE-GANOurs
      PSNR15.183013.223113.979516.134419.357422.3984
      SSIM0.64950.53540.54220.60610.69230.7418
    • Table 4. Quantitative comparison of different methods on test set B

      View table

      Table 4. Quantitative comparison of different methods on test set B

      MetricsGCUDCPLDCPUWCNNFUnIE-GANOurs
      UIQM2.33611.67662.09342.22102.34182.9457
      IE4.22884.46784.54524.36994.65044.6925
      NIQE28.606731.021027.712139.856041.168337.6927
      OG-IQA-0.4489-0.6760-0.7238-0.5479-0.7475-0.8001
    Tools

    Get Citation

    Copy Citation Text

    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

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

    Category: Image Processing

    Received: Oct. 16, 2020

    Accepted: Dec. 8, 2020

    Published Online: Aug. 22, 2021

    The Author Email: Dexing Wang (hcyuan@shou.edu.cn), Hongchun Yuan (hcyuan@shou.edu.cn)

    DOI:10.3788/LOP202158.1610006

    Topics