中文

FUSION:基于频率引导的 underwater 空间图像重建

计算机视觉与模式识别 2025-04-15 v2 人工智能 机器学习 机器人学 图像与视频处理

摘要

水下图像因波长相关的衰减和散射,导致严重的退化,包括颜色失真、可见度降低和结构细节丢失。现有增强方法主要关注空间域处理,忽视了捕获全局颜色分布和长程依赖的频率域潜力。为解决这些限制,我们提出了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)且计算复杂度更低,显示出其适用于实时水下成像应用的潜力。

关键词

引用

@article{arxiv.2504.01243,
  title  = {FUSION: Frequency-guided Underwater Spatial Image recOnstructioN},
  author = {Jaskaran Singh Walia and Shravan Venkatraman and Pavithra LK},
  journal= {arXiv preprint arXiv:2504.01243},
  year   = {2025}
}