AQUA-Net通过频域与光照感知融合,实现高效水下图像增强。
AQUA-Net: Adaptive Frequency Fusion and Illumination Aware Network for Underwater Image Enhancement
- 引入频域与光照双分支,联合优化空间、频率与光照信息。
- 在多个数据集上达到顶尖性能,参数量更少,适合实时应用。
- 适用于复杂深海场景,尤其适合需高鲁棒性的水下视觉任务。
水下图像常因波长相关的光吸收和散射导致严重色彩失真、对比度低和朦胧感。现有深度学习模型计算复杂度高,限制了其在实时水下应用中的部署。本文提出一种新型水下图像增强模型——自适应频域融合与光照感知网络(AQUA-Net),结合残差编码器-解码器结构与双辅助分支,分别在频域和光照域运作。频域融合编码器利用傅里叶域的频率线索丰富空间表征,保留细粒度纹理与结构细节;受Retinex启发,光照感知解码器通过学习的光照图实现自适应曝光校正,分离反射率与光照影响。该联合设计使模型在多样水下条件下有效恢复色彩平衡、视觉对比度与感知真实感。此外,本文构建了一个来自地中海的高分辨率、真实世界水下视频数据集,涵盖具有真实退化特征的深海挑战场景,以支持深度学习模型的稳健评估与开发。大量实验表明,AQUA-Net在多个基准数据集上性能媲美最先进方法,同时参数量更少。消融研究证实频域与光照分支提供互补贡献,提升可见性与色彩表现。整体上,该模型展现出强泛化能力与鲁棒性,为实际水下成像应用提供了有效解决方案。
原文摘要 · Abstract (English)
Underwater images often suffer from severe color distortion, low contrast, and a hazy appearance due to wavelength-dependent light absorption and scattering. Simultaneously, existing deep learning models exhibit high computational complexity, which limits their practical deployment for real-time underwater applications. To address these challenges, this paper presents a novel underwater image enhancement model, called Adaptive Frequency Fusion and Illumination Aware Network (AQUA-Net). It integrates a residual encoder decoder with dual auxiliary branches, which operate in the frequency and illumination domains. The frequency fusion encoder enriches spatial representations with frequency cues from the Fourier domain and preserves fine textures and structural details. Inspired by Retinex, the illumination-aware decoder performs adaptive exposure correction through a learned illumination map that separates reflectance from lighting effects. This joint spatial, frequency, and illumination design enables the model to restore color balance, visual contrast, and perceptual realism under diverse underwater conditions. Additionally, we present a high-resolution, real-world underwater video-derived dataset from the Mediterranean Sea, which captures challenging deep-sea conditions with realistic visual degradations to enable robust evaluation and development of deep learning models. Extensive experiments on multiple benchmark datasets show that AQUA-Net performs on par with SOTA in both qualitative and quantitative evaluations while using less number of parameters. Ablation studies further confirm that the frequency and illumination branches provide complementary contributions that improve visibility and color representation. Overall, the proposed model shows strong generalization capability and robustness, and it provides an effective solution for real-world underwater imaging applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。