Skip to main content
Log in

An underwater attenuation image enhancement method with adaptive color compensation and detail optimization

  • Published:
The Journal of Supercomputing Aims and scope Submit manuscript

Abstract

Aiming at the problems of color compensation bias and loss of detail information in local areas of current underwater image enhancement methods, this paper proposes an underwater attenuation image enhancement method with adaptive color compensation and detail optimization. The method fully considers the attenuation level of each optical channel to guide the color correction based on the attenuation image, and introduces a brightness adjustment method to give the output image a good natural appearance. Gradient-oriented local contrast enhancement and multi-scale edge optimization methods are used to process the color- corrected image separately to obtain two clear images with balanced natural colors, high contrast and good preservation of detail information and then combine with the multi-scale fusion process Artifact-free image fusion is achieved. The experimental results on the UIEB dataset show that the method in this paper improves the UIQM and BRISQUE scores by 14.32% and 8.53%, respectively, and has fewer free parameters, which can effectively enhance attenuated image contrast, detail information and balance image color. In addition, the method in this paper has a complete feature point matching and target detection application, which can be used as a general framework for image preprocessing tasks.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9
Fig. 10
Fig. 11
Fig. 12
Fig. 13
Fig. 14
Fig. 15
Fig. 16

Similar content being viewed by others

Data availability

The data that support the findings of this research are available from the corresponding author upon reasonable request.

Code availability

The code used during the current research is available from the corresponding author on reasonable request.

Abbreviations

ACCDO:

An underwater attenuation image enhancement method with adaptive color compensation and detail optimization

AGCC:

Attenuation image-guided color correction

GOCE:

Gradient-oriented local contrast enhancement

MSEO:

Multi-scale edge optimization

\({I^C}(x,y)\) :

Indicates the input image

\({M^C}(x,y)\) :

Indicates attenuation image

\( \gamma \ \) :

Denotes the gamma transform parameter, taking \( \gamma \ \) = 1.2

\({\phi _{{\mathrm{CC}^*}}}\) :

Indicates the attenuation ratio between color channel

\(I_N^{{C^*}}(x,y)\) :

Indicates color compensated image

\(I_U^C(x,y)\) :

Indicates color equalization image

\(I_{wb}^C(x,y)\) :

Indicates brightness adjustment image

\({\nu ^C}(x,y)\) :

Indicates the brightness adjustment weight

\(\varepsilon \ \) :

Indicates the adjustment speed of each channel’s compensation degree and attenuation change, and takes \(\varepsilon \ \) = 8

\(h\ \) :

Denotes the image chunking ratio of the CLAHE algorithm, taking \(h\ \) = 8

\(I_{{wb - \delta }}^L(x,y)\) :

Indicates CLAHE processed image

\(\nabla I_{{wb - \delta }}^L(x,y)\) :

Denotes the gradient map of the CLAHE processed image

\(n\ \) :

Denotes the neighborhood window size in the gray value mean difference calculation, and takes \(n\ \) = 8

\(\eta \ \) :

Noise judgment threshold, take \(\eta = \overline{\nabla I_{wb - \delta }^L} \left( {x,y} \right) + 25 \)

\(\partial \ \) :

Indicates the local contrast enhancement parameter, taking \(\partial \ \) = 1.2

\({I_{wb - \delta }^L}'(x,y)\) :

Indicates local contrast enhanced image

\(\upsilon \ \) :

Denotes the regularization parameter of the guided filter, taking \(\upsilon = 1 \times {10^{ - 6}}\)

\({I{_{{wb - \delta - g}}^L}}'(x,y)\) :

Denotes the output image after \({I_{wb - \delta }^L}'(x,y)\) -guided filtering process

\(I_{{wb - g}}^C(x,y)\) :

Indicates the output image after \(I_{wb}^C(x,y)\) guided filtering

\(I_{{wb - S}}^C(x,y)\) :

Indicates edge-optimized image

\(g_{{wb - \mathrm{low}}}^C(x,y)\) :

Indicates the low-frequency part of \(I_{wb}^C(x,y)\)

\(g_{{wb - \mathrm{high}}}^C(x,y)\) :

Indicates the high-frequency portion of \(I_{wb}^C(x,y)\)

References

  1. Boudhane M, Nsiri B (2016) Underwater image processing method for fish localization and detection in submarine environment. J Vis Commun Image Represent 39:226–238

    Article  Google Scholar 

  2. Ahn J, Yasukawa S, Sonoda T, Ura T, Ishii K (2017) Enhancement of deep-sea floor images obtained by an underwater vehicle and its evaluation by crab recognition. J Mar Sci Technol 22(4):758–770

    Article  Google Scholar 

  3. Watanabe J-I, Shao Y, Miura N (2019) Underwater and airborne monitoring of marine ecosystems and debris. J Appl Remote Sens 13(4):044509

    Article  Google Scholar 

  4. Peng Y-T, Cosman PC (2017) Underwater image restoration based on image blurriness and light absorption. IEEE Trans Image Process 26(4):1579–1594

    Article  MathSciNet  MATH  Google Scholar 

  5. Peng Y-T, Cao K, Cosman PC (2018) Generalization of the dark channel prior for single image restoration. IEEE Trans Image Process 27(6):2856–2868

    Article  MathSciNet  MATH  Google Scholar 

  6. Kumar M, Bhandari AK (2020) Contrast enhancement using novel white balancing parameter optimization for perceptually invisible images. IEEE Trans Image Process 29:7525–7536

    Article  MATH  Google Scholar 

  7. Wang Y, Zhang J, Cao Y, Wang Z (2017) A deep CNN method for underwater image enhancement. In: 2017 IEEE International Conference on Image Processing (ICIP). IEEE, pp 1382–1386

  8. Li C, Anwar S, Porikli F (2020) Underwater scene prior inspired deep underwater image and video enhancement. Pattern Recognit 98:107038

    Article  Google Scholar 

  9. Wu S, Luo T, Jiang G, Yu M, Xu H, Zhu Z, Song Y (2021) A two-stage underwater enhancement network based on structure decomposition and characteristics of underwater imaging. IEEE J Ocean Eng 46(4):1213–1227

    Article  Google Scholar 

  10. Liu R, Fan X, Hou M, Jiang Z, Luo Z, Zhang L (2018) Learning aggregated transmission propagation networks for haze removal and beyond. IEEE Trans Neural Netw Learn Syst 30(10):2973–2986

    Article  Google Scholar 

  11. Li C, Guo C, Ren W, Cong R, Hou J, Kwong S, Tao D (2019) An underwater image enhancement benchmark dataset and beyond. IEEE Trans Image Process 29:4376–4389

    Article  MATH  Google Scholar 

  12. Fabbri C, Islam MJ, Sattar J (2018) Enhancing underwater imagery using generative adversarial networks. In: 2018 IEEE International Conference on Robotics and Automation (ICRA). IEEE, pp 7159–7165

  13. Zhu J-Y, Park T, Isola P, Efros AA (2017) Unpaired image-to-image translation using cycle-consistent adversarial networks. In: Proceedings of the IEEE International Conference on Computer Vision, pp 2223–2232

  14. Ancuti CO, Ancuti C, De Vleeschouwer C, Bekaert P (2017) Color balance and fusion for underwater image enhancement. IEEE Trans Image Process 27(1):379–393

    Article  MathSciNet  MATH  Google Scholar 

  15. Ancuti CO, Ancuti C, De Vleeschouwer C, Sbert M (2019) Color channel compensation (3c): a fundamental pre-processing step for image enhancement. IEEE Trans Image Process 29:2653–2665

    Article  MATH  Google Scholar 

  16. Tao Y, Dong L, Xu L, Xu W (2021) Effective solution for underwater image enhancement. Opt Express 29(20):32412–32438

    Article  Google Scholar 

  17. Bai L, Zhang W, Pan X, Zhao C (2020) Underwater image enhancement based on global and local equalization of histogram and dual-image multi-scale fusion. IEEE Access 8:128973–128990

    Article  Google Scholar 

  18. Liang Z, Wang Y, Ding X, Mi Z, Fu X (2021) Single underwater image enhancement by attenuation map guided color correction and detail preserved dehazing. Neurocomputing 425:160–172

    Article  Google Scholar 

  19. Zhang W, Dong L, Zhang T, Xu W (2021) Enhancing underwater image via color correction and bi-interval contrast enhancement. Signal Process Image Commun 90:116030

    Article  Google Scholar 

  20. Chang Y, Jung C, Ke P, Song H, Hwang J (2018) Automatic contrast-limited adaptive histogram equalization with dual gamma correction. IEEE Access 6:11782–11792

    Article  Google Scholar 

  21. He K, Sun J, Tang X (2012) Guided image filtering. IEEE Trans Pattern Anal Mach Intell 35(6):1397–1409

    Article  Google Scholar 

  22. Galdran A, Pardo D, Picón A, Alvarez-Gila A (2015) Automatic red-channel underwater image restoration. J Vis Commun Image Represent 26:132–145

    Article  Google Scholar 

  23. Marques TP, Albu AB (2020) L2uwe: a framework for the efficient enhancement of low-light underwater images using local contrast and multi-scale fusion. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, pp 538–539

  24. Song W, Wang Y, Huang D, Tjondronegoro D (2018) A rapid scene depth estimation model based on underwater light attenuation prior for underwater image restoration. In: Pacific Rim Conference on Multimedia. Springer, pp 678–688

  25. Ancuti C, Ancuti CO, Haber T, Bekaert P (2012) Enhancing underwater images and videos by fusion. In: 2012 IEEE Conference on Computer Vision and Pattern Recognition. IEEE, pp 81–88

  26. RECOMMENDATION I-T (1999) Subjective video quality assessment methods for multimedia applications. IEEE Signal Processing Letters (9). IEEE

  27. Wang S, Ma K, Yeganeh H, Wang Z, Lin W (2013) U.s. patent no. 8,429,591. Washington, DC: U.S. patent and trademark office 22(12)

  28. Wang S, Ma K, Yeganeh H, Wang Z, Lin W (2015) A patch-structure representation method for quality assessment of contrast changed images. IEEE Signal Process Lett 22(12):2387–2390 (IEEE)

    Article  Google Scholar 

  29. Yang M, Sowmya A (2015) An underwater color image quality evaluation metric. IEEE Trans Image Process 24(12):6062–6071 (IEEE)

    Article  MathSciNet  MATH  Google Scholar 

  30. Panetta K, Gao C, Agaian S (2015) Human-visual-system-inspired underwater image quality measures. IEEE J Ocean Eng 41(3):541–551 (IEEE)

    Article  Google Scholar 

  31. Mittal A, Moorthy AK, Bovik AC (2012) No-reference image quality assessment in the spatial domain. IEEE Trans Image Process 21(12):4695–4708 (IEEE)

    Article  MathSciNet  MATH  Google Scholar 

  32. Anwar S, Li C (2020) Diving deeper into underwater image enhancement. Signal Process Image Commun 89(11):115978

    Article  Google Scholar 

  33. Hou G, Zhao X, Pan Z, Yang H, Tan L, Li J (2020) Benchmarking underwater image enhancement and restoration, and beyond. IEEE Access 8(10):122078–122091 (IEEE)

    Article  Google Scholar 

  34. Wang Z, Bovik AC, Sheikh HR, Simoncelli EP (2004) Image quality assessment: from error visibility to structural similarity. IEEE Trans Image Process 13(4):600–612 (IEEE)

    Article  Google Scholar 

  35. Wang Z, Bovik AC, Sheikh HR, Simoncelli EP (2011) A feature similarity index for image quality assessment. IEEE Trans Image Process 20(8):2378–2386 (IEEE)

    Article  MathSciNet  Google Scholar 

  36. Liu R, Fan X, Zhu M, Hou M, Luo Z (2020) Real-world underwater enhancement: challenges, benchmarks, and solutions under natural light. IEEE Trans Circuits Syst Video Technol 30(12):4861–4875 (IEEE)

    Article  Google Scholar 

  37. https://github.com/ultralytics/yolov5. Google Scholar

  38. Andrea M, Mario C, Deborah S (2021) Ocean governance and marine spatial planning. UNESCO

Download references

Acknowledgements

We thank the anonymous reviewers for their constructive comments, which have greatly improved the paper.

Funding

This research was supported in part by Guangxi Science and Technology Base and Talent Project (Guike AD18281038), in part by National Natural Science Foundation of China (51805104) in part by Guangxi Natural Science Foundation Program (2018GXNSFBA281184) , in part by Innovation Project of Guangxi Graduate Education(YCSW2022281), in part by Guangxi Innovation Driven Development Special Fund Project of Guangxi Province, China under Grant (AA18118002-3).

Author information

Authors and Affiliations

Authors

Contributions

YP and YY contributed to the conception of the research. YY, GC and BF performed the experiment. YP and GC contributed significantly to analysis and manuscript preparation. YP, YY and XG performed the data analyses and wrote the manuscript. YP. YY and XG provided funding. All authors helped perform the analysis with constructive discussions.

Corresponding author

Correspondence to Xingyu Gao.

Ethics declarations

Conflict of interest

The authors declare that they have no conflict of interest.

Consent to participate

Informed consent was obtained from all individual participants included in the research.

Consent for publication

All authors of this research have given their consent for publication.

Ethics approval

This article does not contain any researches with human participants or animals performed by any of the authors.

Additional information

Publisher's Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Peng, Y., Yan, Y., Chen, G. et al. An underwater attenuation image enhancement method with adaptive color compensation and detail optimization. J Supercomput 79, 1544–1570 (2023). https://doi.org/10.1007/s11227-022-04720-z

Download citation

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s11227-022-04720-z

Keywords

Navigation