Adaptive Underwater Image Enhancement Guided by Degradation-Perception Priors
-
摘要: 针对传统水下图像增强方法中非物理模型方法缺乏退化机理约束、易产生欠增强或过增强, 以及物理模型方法关键参数估计稳定性受限的问题, 文中提出一种基于退化感知先验引导的自适应水下图像增强方法。该方法以图像局部统计特征为退化表征, 构建具有物理解释性的退化感知先验, 从而在不依赖目标数据集网络训练、且不显式求解完整水下成像模型参数的前提下, 保持非物理模型方法处理的灵活性并为增强过程提供退化机理约束。首先, 通过最小衰减通道策略校正色偏, 并基于引导滤波估计图像的非均匀环境光场实现亮度自适应均衡。随后, 针对水下退化规律将图像的局部对比度方差、环境光场偏移、局部能量统计和梯度幅值分别映射为雾度分布、曝光风险、噪声分布和结构显著性四种退化感知先验, 引导局部去雾和多尺度纹理增强。实验结果表明, 该方法在校正色偏、结构保持和细节纹理增强方面具有较好表现, 并在运行效率上有一定优势。Abstract: To address the limitations of traditional underwater image enhancement methods, where non-physical-model-based approaches often lack degradation-mechanism constraints and may cause under-enhancement or over-enhancement, while physical-model-based approaches rely on stable estimation of key parameters, this paper proposes an adaptive enhancement method guided by degradation-aware priors. The method constructs physically interpretable priors from local statistical image features, thereby preserving the flexibility of non-physical-model-based enhancement while introducing degradation constraints, without requiring target-dataset network training or explicit inversion of a complete underwater imaging model. Specifically, color deviation is first corrected using a minimum attenuation channel strategy, and adaptive luminance equalization is achieved by estimating a non-uniform ambient light field via guided filtering. Local contrast variance, ambient light field deviation, local energy statistics, and gradient magnitude are then mapped to haze distribution, exposure risk, noise distribution, and structural saliency, respectively, to guide local dehazing and multi-scale texture enhancement. Experiments show that the proposed method performs favorably in color deviation, structural preservation, texture detail enhancement, and computational efficiency.
-
Key words:
- underwater image enhancement /
- degradation perception /
- prior-guided /
- adaptive
-
表 1 不同算法测试指标
Table 1. Test indicators of different algorithms
算法 PSNR SSIM UIQM UCIQE 处理时间 UDCP 13.65 0.62 2.25 0.54 44.61 s MLLE 19.54 0.83 2.86 0.63 2.76 s PCDE 15.82 0.71 2.53 0.59 2.82 s CBLA 16.12 0.67 2.97 0.68 0.57 s FiveA+ 20.34 0.80 3.14 0.54 2.37 s Osmosis 19.82 0.78 3.01 0.57 8 h 37 min UDNet 19.68 0.82 3.08 0.52 6.93 s HUPE 21.62 0.85 3.26 0.74 5.22 s 文中算法 23.15 0.88 3.52 0.72 1.36 s 表 2 模块消融实验测试指标
Table 2. Test indicators of modular ablation experiment
方法 PSNR SSIM UIQM UCIQE w/o M1 17.65 0.75 2.05 0.53 w/o M2 19.32 0.81 2.43 0.61 w/o M3 21.13 0.69 2.78 0.65 完整方法 23.15 0.88 3.52 0.72 表 3 退化感知先验消融实验测试指标
Table 3. Test indicators of degradation perception prior ablation experiment
方法 PSNR SSIM UIQM UCIQE M3 输入 21.13 0.69 2.78 0.65 w/o H,E 21.80 0.85 3.02 0.68 w/o N,G 22.45 0.76 3.40 0.70 完整H,E,N,G 23.15 0.88 3.52 0.72 -
[1] Ma Z Y, Oh C. A wavelet-based dual-stream network for underwater image enhancement[C]//2022 IEEE International Conference on Acoustics, Speech and Signal Processing(ICASSP), 2022: 2769-2773. [2] Muniraj M, Dhandapani V. Underwater image enhancement by modified color correction and adaptive look-up table with edge-preserving filter[J]. Signal Processing: Image Communication, 2023, 113: 116939. doi: 10.1016/j.image.2023.116939 [3] Iqbal K, Salam R A, Osman A, et al. Underwater image enhancement using an integrated colour model[J]. IAENG International Journal of Computer Science, 2007, 34(2): 239-244. [4] Iqbal K, Odetayo M, James A, et al. Enhancing the low quality images using unsupervised colour correction method[C]//2010 IEEE International Conference on Systems, Man and Cybernetics, 2010: 1703-1709. [5] Ulutas G, Ustubioglu B. Underwater image enhancement using contrast limited adaptive histogram equalization and layered difference representation[J]. Multimedia Tools and Applications, 2021, 80(10): 15067-15091. doi: 10.1007/s11042-020-10426-2 [6] Yan X, Wang G, Wang G, et al. A novel biologically-inspired method for underwater image enhancement[J]. Signal Processing: Image Communication, 2022, 104: 116670. doi: 10.1016/j.image.2022.116670 [7] Zhuang P X, Ding X H. Underwater image enhancement using an edge-preserving filtering Retinex algorithm[J]. Multimedia Tools and Applications, 2020, 79: 17257-17277. doi: 10.1007/s11042-019-08404-4 [8] Zhuang P X, Ding X H. Underwater image enhancement using an edge-preserving filtering Retinex algorithm[J]. Multimedia Tools and Applications, 2020, 79(25): 17257-17277. doi: 10.1007/s11042-020-08739-3 [9] Zhou J, Yang T, Zhang W S. Underwater vision enhancement technologies: a comprehensive review, challenges, and recent trends[J]. Applied Intelligence, 2023, 53(3): 3594-3621. doi: 10.1007/s10489-022-03767-y [10] 丁雪妍. 基于物理模型与深度学习的水下光学图像增强方法研究[D]. 大连: 大连海事大学, 2022: 25-38. [11] 周妍, 顾鑫涛, 李庆武. 基于背景光修正成像模型的水下图像复原[J]. 电子与信息学报, 2022, 44(10): 3363-3371.Zhou Y, Gu X T, Li Q W. Underwater image restoration based on modified imaging model with background light correction[J]. Journal of Electronics & Information Technology, 2022, 44(10): 3363-3371. [12] 林森, 迟凯晨, 唐延东. 基于复原结构与增强纹理融合的水下图像清晰化[J]. 控制与决策, 2022, 37(3): 635-644.Lin S, Chi K C, Tang Y D. Underwater image clarification based on fusion of restored structure and enhanced texture[J]. Control and Decision, 2022, 37(3): 635-644. [13] Zhuang P X, Sun H H, Li G P, et al. Underwater image enhancement via minimal color loss and locally adaptive contrast enhancement[J]. IEEE Transactions on Image Processing, 2022, 31: 3997-4010. doi: 10.1109/TIP.2022.3177129 [14] Chang B C, Li J J, Wang H Y, et al. Attention-based color consistency underwater image enhancement network[J]. Signal, Image and Video Processing, 2024, 18(5): 4385-4394. doi: 10.1007/s11760-024-03080-w [15] Wang Z, Shen L, Xu M, et al. Domain adaptation for underwater image enhancement[J]. IEEE Transactions on Image Processing, 2023, 32: 1442-1457. doi: 10.1109/TIP.2023.3244647 [16] Tolie H F, Ren J, Elyan E. DICAM: deep inception and channel-wise attention modules for underwater image enhancement[J]. Neurocomputing, 2024, 584: 127585. doi: 10.1016/j.neucom.2024.127585 [17] Fu Z Q, Wang W, Huang Y, et al. Uncertainty inspired underwater image enhancement[C]//17th European Conference on Computer Vision, 2022: 465-482. [18] Peng L, Zhu C, Bian L. U-shape transformer for underwater image enhancement[J]. IEEE Transactions on Image Processing, 2023, 32: 3066-3079. doi: 10.1109/TIP.2023.3276332 [19] Pramanick A, Sarma S, Sur A. X-CAUNET: cross-color channel attention with underwater image-enhancing transformer[C]//2024 IEEE International Conference on Acoustics, Speech and Signal Processing(ICASSP), 2024: 3550-3554. [20] Xu T, Xu S, Chen X, et al. Multi-core token mixer: a novel approach for underwater image enhancement[J]. Machine Vision and Applications, 2025, 36(2): 37. doi: 10.1007/s00138-024-01651-y [21] Li Z, Yan K X, Zhou D M, et al. A novel highland and freshwater-circumstance dataset: advancing underwater image enhancement[J]. The Visual Computer, 2024, 40(10): 7471-7489. doi: 10.1007/s00371-024-03285-7 [22] Xu S H, Wang J H, He N, et al. Underwater image enhancement method based on a cross attention mechanism[J]. Multimedia Systems, 2024, 30(1): 26. doi: 10.1007/s00530-023-01224-5 [23] Wang Z, Li C Y, Mo Y, et al. RCA-CycleGAN: unsupervised underwater image enhancement using red channel attention optimized CycleGAN[J]. Displays, 2023, 76: 102359. doi: 10.1016/j.displa.2022.102359 [24] YAN H, ZHANG Z, XU J, et al. UW-CycleGAN: model-driven CycleGAN for underwater image restoration[J]. IEEE Transactions on Geoscience and Remote Sensing, 2023, 61: 1-7. doi: 10.1109/tgrs.2023.3315772 [25] Saleh A, Sheaves M, Jerry D, et al. Adaptive deep learning framework for robust unsupervised underwater image enhancement[J]. Expert Systems with Applications, 2025, 268: 126314. [26] Qiao N, Sun C, Dong L, et al. Semi-supervised feature distillation and unsupervised domain adversarial distillation for underwater image enhancement[J]. IEEE Transactions on Circuits and Systems for Video Technology, 2024, 34(8): 7671-7682. [27] Zhang Z, Jiang Z, Ma L, et al. HUPE: heuristic underwater perceptual enhancement with semantic collaborative learning[J]. International Journal of Computer Vision, 2025, 133(6): 3259-3277. [28] Huang Y, Yuan F, Xiao F, et al. Underwater image enhancement based on zero-reference deep network[J]. IEEE Journal of Oceanic Engineering, 2023, 48(3): 903-924. [29] Nathan O B, Levy D, Treibitz T, et al. Osmosis: RGBD diffusion prior for underwater image restoration[C]//18th European Conference on Computer Vision, 2024: 302-319. [30] HE K M, SUN J, TANG X O. Guided image filtering[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2013, 35(6): 1397-1409. [31] Drews P L, Nascimento E R, Botelho S S, et al. Underwater depth estimation and image restoration based on single images[J]. IEEE Computer Graphics and Applications, 2016, 36(2): 24-35. [32] Zhang W D, Jin S Y, Zhuang P X, et al. Underwater image enhancement via piecewise color correction and dual prior optimized contrast enhancement[J]. IEEE Signal Processing Letters, 2023, 30: 229-233. [33] Jha M, Bhandari A K. CBLA: color-balanced locally adjustable underwater image enhancement[J]. IEEE Transactions on Instrumentation and Measurement, 2024, 73: 1-11. [34] Jiang J, Ye T, Bai J, et al. Five A+ network: you only need 9K parameters for underwater image enhancement[PP/OL]. arXiv (2023-05-15)[2026-02-07]. https://doi.org/10.48550/arXiv.2305.08824. -

下载: