基于暗通道先验与自适应物理模型补偿的水下图像增强方法
Underwater Image Enhancement Method Based on Dark Channel Prior and Adaptive Physical Model Compensation
摘要: 水下图像在海洋探测、资源勘查与工程运维中应用价值显著,但受水体光吸收与散射作用影响,易出现色偏、对比度低、局部照度不足等质量退化问题。本文提出一种融合暗通道先验与自适应物理模型补偿的水下图像增强方法,先依据水下光波长衰减特性与暗通道先验掩码补偿RGB通道偏差,再通过自适应亮度区间的非线性拉伸算法,实现全局和局部对比度增强,最后以最亮通道为补偿源,结合自适应暗通道掩码实现低照度区域增强。与深度学习方法相比,本算法物理建模明确、可解释性强,在UIEB数据集验证结果表明,SSIM、UIQM、UCIQE指标分别为0.8986、2.8984、0.6283,在Orin系列开发板上处理速率满足工业标准,形成了一套适用于工程应用的水下图像增强处理工具,可有效支撑潜航器下游感知任务性能提升。
Abstract: Underwater images present significant application value in marine exploration, resource exploration and engineering operation and maintenance. However, due to the absorption and scattering of light by water bodies, they are prone to quality degradation such as color cast, low contrast and insufficient local illumination. This paper proposes an underwater image enhancement method that integrates dark channel prior and adaptive physical model compensation. Firstly, the RGB channel deviation is compensated based on the attenuation characteristics of underwater light wavelengths and the dark channel prior mask. Then, a nonlinear stretching algorithm with adaptive brightness intervals is adopted to achieve global and local contrast enhancement. Finally, with the brightest channel as the compensation source, the enhancement of low-illumination regions is realized combined with the adaptive dark channel mask. Compared with deep learning-based methods, the proposed algorithm has explicit physical modeling and strong interpretability. The verification results on the UIEB dataset show that the SSIM, UIQM and UCIQE metrics are 0.8986, 2.8984 and 0.6283 respectively. The processing speed on Orin series development boards meets industrial standards, and a complete set of underwater image enhancement processing tools suitable for engineering applications is developed, which can effectively improve the performance of downstream perception tasks for underwater vehicles.
文章引用:陈自豪, 田野, 袁小军, 徐振森, 袁希文, 张慧源, 李晨. 基于暗通道先验与自适应物理模型补偿的水下图像增强方法[J]. 人工智能与机器人研究, 2026, 15(3): 913-927. https://doi.org/10.12677/airr.2026.153084

参考文献

[1] Hsieh, Y.Z. and Chang, M.C. (2025) Underwater Image Enhancement and Attenuation Restoration Based on Depth and Backscatter Estimation. IEEE Transactions on Computational Imaging, 11, 321-332. [Google Scholar] [CrossRef
[2] Zhu, J., Wang, H., Chen, Z., Zhang, L. and Zhang, M. (2025) Underwater Image Enhancement through Color Deviation Detection-Guided Peak Flattening. Signal, Image and Video Processing, 19, Article No. 8. [Google Scholar] [CrossRef
[3] Saoud, L.S., Elmezain, M., Sultan, A., et al. (2024) Seeing through the Haze: A Comprehensive Review of Underwater Image Enhancement Techniques. IEEE Access, 12, 145206-145233. [Google Scholar] [CrossRef
[4] Cong, X., Zhao, Y., Gui, J., et al. (2024) A Comprehensive Survey on Underwater Image Enhancement Based on Deep Learning.
https://arxiv.org/abs/2405.19684
[5] Drews Jr, P., do Nascimento, E., Moraes, F., Botelho, S. and Campos, M. (2013) Transmission Estimation in Underwater Single Images. 2013 IEEE International Conference on Computer Vision Workshops, Sydney, 2-8 December 2013, 825-830. [Google Scholar] [CrossRef
[6] Dhal, K.G., Das, A., Ray, S., Gálvez, J. and Das, S. (2021) Histogram Equalization Variants as Optimization Problems: A Review. Archives of Computational Methods in Engineering, 28, 1471-1496. [Google Scholar] [CrossRef
[7] Qing, Y., Wang, Y., Yan, H., Xie, X. and Wu, Z. (2024) Unformer: A Transformer-Based Approach for Adaptive Multiscale Feature Aggregation in Underwater Image Enhancement. IEEE Transactions on Artificial Intelligence, 6, 1024-1037. [Google Scholar] [CrossRef
[8] Panda, G., Kundu, S., Bhattacharya, S. and Routray, A. (2025) SINET: Sparsity-Driven Interpretable Neural Network for Underwater Image Enhancement. 2025 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), Hyderabad, 6-11 April 2025, 1-5. [Google Scholar] [CrossRef
[9] Hitam, M.S., Yussof, W.N.J.H.W., Awalludin, E.A. and Bachok, Z. (2013) Mixture Contrast Limited Adaptive Histogram Equalization for Underwater Image Enhancement. 2013 International Conference on Computer Applications Technology (ICCAT), Sousse, 20-22 January 2013, 1-5. [Google Scholar] [CrossRef
[10] He, K., Sun, J. and Tang, X. (2010) Single Image Haze Removal Using Dark Channel Prior. IEEE Transactions on Pattern Analysis and Machine Intelligence, 33, 2341-2353. [Google Scholar] [CrossRef] [PubMed]
[11] Chiang, J.Y. and Chen, Y.C. (2011) Underwater Image Enhancement by Wavelength Compensation and Dehazing. IEEE Transactions on Image Processing, 21, 1756-1769. [Google Scholar] [CrossRef] [PubMed]
[12] Peng, Y.T. and Cosman, P.C. (2017) Underwater Image Restoration Based on Image Blurriness and Light Absorption. IEEE Transactions on Image Processing, 26, 1579-1594. [Google Scholar] [CrossRef] [PubMed]
[13] Li, C., Quo, J., Pang, Y., Chen, S. and Wang, J. (2016) Single Underwater Image Restoration by Blue-Green Channels Dehazing and Red Channel Correction. 2016 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), Shanghai, 20-25 March 2016, 1731-1735. [Google Scholar] [CrossRef
[14] Abdul Ghani, A.S. and Mat Isa, N.A. (2014) Underwater Image Quality Enhancement through Composition of Dual-Intensity Images and Rayleigh-Stretching. SpringerPlus, 3, Article No. 757. [Google Scholar] [CrossRef] [PubMed]
[15] Li, C., Guo, C., Ren, W., Cong, R., Hou, J., Kwong, S., et al. (2019) An Underwater Image Enhancement Benchmark Dataset and Beyond. IEEE Transactions on Image Processing, 29, 4376-4389. [Google Scholar] [CrossRef] [PubMed]
[16] Peng, L., Zhu, C. and Bian, L. (2023) U-Shape Transformer for Underwater Image Enhancement. IEEE Transactions on Image Processing, 32, 3066-3079. [Google Scholar] [CrossRef] [PubMed]
[17] Wang, Z., Bovik, A.C., Sheikh, H.R., et al. (2004) Image Quality Assessment: From Error Visibility to Structural Similarity. IEEE Transactions on Image Processing, 13, 600-612. [Google Scholar] [CrossRef] [PubMed]
[18] Zhang, L., Zhang, L., Mou, X., et al. (2011) FSIM: A Feature Similarity Index for Image Quality Assessment. IEEE Transactions on Image Processing, 20, 2378-2386. [Google Scholar] [CrossRef] [PubMed]
[19] Panetta, K., Gao, C. and Agaian, S. (2015) Human-Visual-System-Inspired Underwater Image Quality Measures. IEEE Journal of Oceanic Engineering, 41, 541-551. [Google Scholar] [CrossRef
[20] Yang, M. and Sowmya, A. (2015) An Underwater Color Image Quality Evaluation Metric. IEEE Transactions on Image Processing, 24, 6062-6071. [Google Scholar] [CrossRef] [PubMed]
[21] Liu, R., Jiang, Z., Yang, S. and Fan, X. (2022) Twin Adversarial Contrastive Learning for Underwater Image Enhancement and Beyond. IEEE Transactions on Image Processing, 31, 4922-4936. [Google Scholar] [CrossRef] [PubMed]
[22] Shen, Z., Xu, H., Luo, T., Song, Y. and He, Z. (2023) UDAformer: Underwater Image Enhancement Based on Dual Attention Transformer. Computers & Graphics, 111, 77-88. [Google Scholar] [CrossRef
[23] Jiang, J., Ye, T., Bai, J., et al. (2023) Five A+ Network: You Only Need 9K Parameters for Underwater Image Enhancement.
https://arxiv.org/abs/2305.08824
[24] Zhang, G., Li, C., Yan, J. and Zheng, Y. (2024) ULD-CycleGAN: An Underwater Light Field and Depth Map-Optimized CycleGAN for Underwater Image Enhancement. IEEE Journal of Oceanic Engineering, 49, 1275-1288. [Google Scholar] [CrossRef