arXiv:2504.01243cs.CVcs.AI2025-04CVPR被引 7

融合频域与空间域信息,提升水下图像重建质量。

FUSION: Frequency-guided Underwater Spatial Image recOnstructioN

  • 双域协同:空间域用多尺度卷积,频域用FFT提取全局结构。
  • 在UIEB数据集上达到23.717dB的最高PSNR和0.883的最高SSIM。
  • 参数仅0.28M,适合实时水下成像,适合嵌入式设备部署。

水下图像受波长依赖性衰减和散射影响,普遍存在色彩失真、能见度降低和结构细节丢失问题。现有增强方法主要聚焦空间域处理,忽视了频域在捕捉全局色彩分布和长程依赖方面的潜力。为此,我们提出FUSION,一种联合利用空间与频域信息的深度学习框架。FUSION在空间域通过多尺度卷积核与自适应注意力机制独立处理各RGB通道,同时在频域通过基于FFT的频率注意力提取全局结构信息。频率引导融合模块整合两域互补特征,再经通道间融合与自适应重校准,确保色彩平衡。在基准数据集UIEB、EUVP、SUIM-E上的大量实验表明,FUSION在重建保真度(UIEB上最高PSNR达23.717 dB,SSIM达0.883)、感知质量(UIEB上最低LPIPS为0.112)和视觉增强指标(UIEB上最佳UIQM为3.414)方面均优于现有方法,且参数量仅0.28M,计算复杂度低,适用于实时水下成像应用。

原文摘要 · Abstract (English)

Underwater images suffer from severe degradations, including color distortions, reduced visibility, and loss of structural details due to wavelength-dependent attenuation and scattering. Existing enhancement methods primarily focus on spatial-domain processing, neglecting the frequency domain's potential to capture global color distributions and long-range dependencies. To address these limitations, we propose FUSION, a dual-domain deep learning framework that jointly leverages spatial and frequency domain information. FUSION independently processes each RGB channel through multi-scale convolutional kernels and adaptive attention mechanisms in the spatial domain, while simultaneously extracting global structural information via FFT-based frequency attention. A Frequency Guided Fusion module integrates complementary features from both domains, followed by inter-channel fusion and adaptive channel recalibration to ensure balanced color distributions. Extensive experiments on benchmark datasets (UIEB, EUVP, SUIM-E) demonstrate that FUSION achieves state-of-the-art performance, consistently outperforming existing methods in reconstruction fidelity (highest PSNR of 23.717 dB and SSIM of 0.883 on UIEB), perceptual quality (lowest LPIPS of 0.112 on UIEB), and visual enhancement metrics (best UIQM of 3.414 on UIEB), while requiring significantly fewer parameters (0.28M) and lower computational complexity, demonstrating its suitability for real-time underwater imaging applications.

水下图像双域融合频域增强实时处理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。