Laser & Optoelectronics Progress, Volume. 60, Issue 16, 1610007(2023)

Underwater Image Restoration Based on Total Variation and Color Balance

Yali Hao, Guojia Hou*, Yuemei Li, Baoxiang Huang, and Zhenkuan Pan
Author Affiliations
  • College of Computer Science & Technology, Qingdao University, Qingdao 266071, Shandong, China
  • show less
    Figures & Tables(10)
    General framework of proposed method
    An example of estimating global background light. (a) Original image; (b) result of the quad-tree subdivision; (c) estimated global background light in red rectangle
    Comparison of the estimated transmission maps using different methods. (a) Original image; (b)-(d) the transmission maps obtained by DCP, UDCP, and the proposed method; (e)-(g) the corresponding restored results by DCP, UDCP, and the proposed method
    Restoration results of different algorithms. (a) Original images; (b) RCP algorithm; (c) IBLA algorithm; (d) ULAP algorithm;(e) UNTV algorithm; (f) UWCNN algorithm; (g) Bayes algorithm; (h) proposed algorithm
    Comparison of color correction. (a) Color checker; (b) original image; (c) RCP algorithm;(d) IBLA algorithm;(e) ULAP algorithm; (f) UNTV algorithm; (g) UWCNN algorithm; (h) Bayes algorithm; (i) proposed algorithm
    Ablation experiment. (a) Original images; (b) only defogging and deblurring; (c) only color correction; (d) proposed algorithm
    • Table 1. Comparison of UCIQE metrics of restored images with different algorithms

      View table

      Table 1. Comparison of UCIQE metrics of restored images with different algorithms

      No.RawRCPIBLAULAPUNTVUWCNNBayesProposed algorithm
      image10.45590.52600.56730.63180.61820.43610.50230.6375
      image20.56630.57830.61960.65020.63500.50670.50690.6690
      image30.50080.54160.58650.59600.62970.46220.49090.7836
      image40.41750.53370.47780.49440.58980.45310.41030.5971
      image50.45260.52720.53660.55010.62040.44660.51010.6319
      image60.58540.57250.64520.47880.62690.47440.44170.6465
    • Table 2. Comparison of UIQM metrics of restored images with different algorithms

      View table

      Table 2. Comparison of UIQM metrics of restored images with different algorithms

      No.RawRCPIBLAULAPUNTVUWCNNBayesProposed algorithm
      image11.20101.42311.44091.60841.64361.19261.31371.6701
      image21.14161.22801.34311.47361.59941.03120.98691.6736
      image31.38311.47891.57521.56901.69371.30321.31741.7225
      image41.42581.60141.58111.68961.61381.36711.27821.6931
      image51.19041.50811.40571.43891.70191.26541.31861.7652
      image61.77421.82491.80801.64881.89681.49760.94661.9033
    • Table 3. Comparison of FDUM metrics of restored images with different algorithms

      View table

      Table 3. Comparison of FDUM metrics of restored images with different algorithms

      No.RawRCPIBLAULAPUNTVUWCNNBayesProposed algorithm
      image10.46130.59190.70310.96751.01010.37990.52501.0702
      image20.53490.59460.66210.76101.00620.34690.37121.0318
      image30.51520.60490.73200.73151.04990.40090.47351.0546
      image40.34430.47230.50450.52660.76960.36030.29280.7730
      image50.45960.64070.72160.74631.10970.42610.58741.1893
      image60.57830.57680.83280.52630.95320.33360.31461.0874
    • Table 4. Comparison of average values of different objective quality evaluation metrics

      View table

      Table 4. Comparison of average values of different objective quality evaluation metrics

      MetricRawRCPIBLAULAPUNTVUWCNNBayesProposed algorithm
      UCIQE0.51210.56290.59690.60530.59780.57390.51840.6237
      UIQM1.21471.39951.44681.411.61471.37281.04671.6546
      FDUM0.46120.53930.66840.65750.87300.55520.50180.8845
    Tools

    Get Citation

    Copy Citation Text

    Yali Hao, Guojia Hou, Yuemei Li, Baoxiang Huang, Zhenkuan Pan. Underwater Image Restoration Based on Total Variation and Color Balance[J]. Laser & Optoelectronics Progress, 2023, 60(16): 1610007

    Download Citation

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

    Category: Image Processing

    Received: Sep. 1, 2022

    Accepted: Oct. 18, 2022

    Published Online: Aug. 15, 2023

    The Author Email: Hou Guojia (hgj2015@qdu.edu.cn)

    DOI:10.3788/LOP222442

    Topics