• Laser & Optoelectronics Progress
  • Vol. 58, Issue 12, 1210024 (2021)
Wei Song*, Jingjing Xing, Yanling Du, and Qi He**
Author Affiliations
  • Department of Information and Technology, Shanghai Ocean University, Shanghai 201306, China
  • show less
    DOI: 10.3788/LOP202158.1210024 Cite this Article Set citation alerts
    Wei Song, Jingjing Xing, Yanling Du, Qi He. Underwater Image Enhancement Based on Generative Adversarial Network with Preprocessed Image Penalty[J]. Laser & Optoelectronics Progress, 2021, 58(12): 1210024 Copy Citation Text show less
    Framework of our algorithm
    Fig. 1. Framework of our algorithm
    Images processed by different stretching algorithms. (a) Original image; (b) image processed by traditional histogram stretching algorithm; (c) image processed by improved histogram stretching algorithm improved histogram stretching algorithm
    Fig. 2. Images processed by different stretching algorithms. (a) Original image; (b) image processed by traditional histogram stretching algorithm; (c) image processed by improved histogram stretching algorithm improved histogram stretching algorithm
    Structure of the GAN. (a) Generator; (b) convolution block; (c) discriminator
    Fig. 3. Structure of the GAN. (a) Generator; (b) convolution block; (c) discriminator
    Enhancement results of different methods on underwater images. (a) Original image; (b) HE; (c) RGHS; (d) UDCP; (e) IBLA; (f) Fusion; (g) Water-Net; (h) Uresnet; (i) FunieGAN; (j) UGAN; (k) ours
    Fig. 4. Enhancement results of different methods on underwater images. (a) Original image; (b) HE; (c) RGHS; (d) UDCP; (e) IBLA; (f) Fusion; (g) Water-Net; (h) Uresnet; (i) FunieGAN; (j) UGAN; (k) ours
    Test results of different methods on the synthetic data set. (a) Orginal image; (b) Water-Net; (c) Uresnet; (d) FunieGAN; (e) UGAN; (f) ours
    Fig. 5. Test results of different methods on the synthetic data set. (a) Orginal image; (b) Water-Net; (c) Uresnet; (d) FunieGAN; (e) UGAN; (f) ours
    Detail retention capabilities of different methods. (a) Original image; (b) details of the solid line frame; (c) details of the dotted line frame; (d) details of the dotted frame
    Fig. 6. Detail retention capabilities of different methods. (a) Original image; (b) details of the solid line frame; (c) details of the dotted line frame; (d) details of the dotted frame
    Detection results of underwater targets. (a) Original image; (b) enhanced image
    Fig. 7. Detection results of underwater targets. (a) Original image; (b) enhanced image
    TypeMethodPSNR /dBSSIM
    Model-based methodHE16.280.781
    RGHS19.490.832
    Fusion22.120.846
    UDCP12.640.578
    IBLA12.420.463
    Deeplearning-based methodWater-Net19.120.813
    Uresnet17.660.740
    FUnieGAN17.630.761
    UGAN19.600.840
    Ours20.590.844
    Table 1. PSNR and SSIM of different methods on the UIEBD test set
    TypeMethodUIQMUICMUISMUIConM
    Model-based methodHE2.90611.1277.5540.101
    RGHS2.1697.4835.7960.069
    Fusion2.9228.1778.0040.091
    UDCP1.9787.7115.6090.029
    IBLA1.7188.1154.4740.046
    Deep learning-based methodWater-Net3.0305.0147.5450.184
    Uresnet2.8845.1936.8220.202
    FUnieGAN2.7826.1217.0210.149
    UGAN3.4235.9888.3070.235
    Ours3.4715.4838.4130.232
    Table 2. UIQM of different methods in EUVP data set
    DatasetMethodPSNR /dBSSIM
    EUVPwithout penalty19.710.833
    with penalty20.590.844
    Synthesiswithout penalty15.250.709
    with penalty15.600.716
    Table 3. Influence of penalty items on the image enhancement effect
    Data setMethodPSNR /dBSSIM
    EUVPUDCP18.650.832
    CycleGAN19.930.843
    Ours20.590.844
    SynthesisUDCP14.190.687
    CycleGAN15.370.711
    Ours15.600.716
    Table 4. Influence of different preprocessing methods on the image enhancement effect
    Wei Song, Jingjing Xing, Yanling Du, Qi He. Underwater Image Enhancement Based on Generative Adversarial Network with Preprocessed Image Penalty[J]. Laser & Optoelectronics Progress, 2021, 58(12): 1210024
    Download Citation