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.
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
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
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
Watanabe J-I, Shao Y, Miura N (2019) Underwater and airborne monitoring of marine ecosystems and debris. J Appl Remote Sens 13(4):044509
Peng Y-T, Cosman PC (2017) Underwater image restoration based on image blurriness and light absorption. IEEE Trans Image Process 26(4):1579–1594
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
Kumar M, Bhandari AK (2020) Contrast enhancement using novel white balancing parameter optimization for perceptually invisible images. IEEE Trans Image Process 29:7525–7536
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
Li C, Anwar S, Porikli F (2020) Underwater scene prior inspired deep underwater image and video enhancement. Pattern Recognit 98:107038
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
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
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
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
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
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
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
Tao Y, Dong L, Xu L, Xu W (2021) Effective solution for underwater image enhancement. Opt Express 29(20):32412–32438
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
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
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
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
He K, Sun J, Tang X (2012) Guided image filtering. IEEE Trans Pattern Anal Mach Intell 35(6):1397–1409
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
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
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
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
RECOMMENDATION I-T (1999) Subjective video quality assessment methods for multimedia applications. IEEE Signal Processing Letters (9). IEEE
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)
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)
Yang M, Sowmya A (2015) An underwater color image quality evaluation metric. IEEE Trans Image Process 24(12):6062–6071 (IEEE)
Panetta K, Gao C, Agaian S (2015) Human-visual-system-inspired underwater image quality measures. IEEE J Ocean Eng 41(3):541–551 (IEEE)
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)
Anwar S, Li C (2020) Diving deeper into underwater image enhancement. Signal Process Image Commun 89(11):115978
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)
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)
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)
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)
https://github.com/ultralytics/yolov5. Google Scholar
Andrea M, Mario C, Deborah S (2021) Ocean governance and marine spatial planning. UNESCO
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
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
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
About this article
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
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11227-022-04720-z