轻量级水下图像增强网络,兼顾效果与实时性
WWE-UIE: A Wavelet & White Balance Efficient Network for Underwater Image Enhancement
- 融合自适应白平衡、小波分解与梯度感知模块
- 参数量和计算量大幅降低,支持实时推理
- 适合嵌入式设备或移动端的水下视觉应用
水下图像增强旨在恢复因波长依赖性吸收和散射造成的可见度下降与颜色失真。近年来,结合领域先验与现代深度神经网络的混合方法虽取得优异性能,但计算开销高,难以在实时场景中应用。本文提出WWE-UIE,一种紧凑高效的增强网络,集成三种可解释先验:首先,自适应白平衡缓解强波长依赖性色彩衰减,尤其抑制蓝绿色主导问题;其次,基于小波的增强模块(WEB)实现多频带分解,使网络能同时捕捉全局结构与精细纹理;第三,梯度感知模块(SGFB)利用可学习门控的Sobel算子,显式保留散射导致退化的边缘结构。在基准数据集上的大量实验表明,WWE-UIE在参数量和浮点运算次数(FLOPs)显著减少的前提下,仍达到竞争力的复原质量,支持资源受限平台的实时推理。消融实验与可视化进一步验证各组件的有效性。源代码见https://github.com/chingheng0808/WWE-UIE。
原文摘要 · Abstract (English)
Underwater Image Enhancement (UIE) aims to restore visibility and correct color distortions caused by wavelength-dependent absorption and scattering. Recent hybrid approaches, which couple domain priors with modern deep neural architectures, have achieved strong performance but incur high computational cost, limiting their practicality in real-time scenarios. In this work, we propose WWE-UIE, a compact and efficient enhancement network that integrates three interpretable priors. First, adaptive white balance alleviates the strong wavelength-dependent color attenuation, particularly the dominance of blue-green tones. Second, a wavelet-based enhancement block (WEB) performs multi-band decomposition, enabling the network to capture both global structures and fine textures, which are critical for underwater restoration. Third, a gradient-aware module (SGFB) leverages Sobel operators with learnable gating to explicitly preserve edge structures degraded by scattering. Extensive experiments on benchmark datasets demonstrate that WWE-UIE achieves competitive restoration quality with substantially fewer parameters and FLOPs, enabling real-time inference on resource-limited platforms. Ablation studies and visualizations further validate the contribution of each component. The source code is available at https://github.com/chingheng0808/WWE-UIE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。