针对水下图像退化问题,提出自适应增强新方法。
DACA-Net: A Degradation-Aware Conditional Diffusion Network for Underwater Image Enhancement
- 根据退化程度生成连续评分,指导扩散模型自适应修复
- 在多个基准数据集上提升颜色保真度与结构细节表现
- 适合需要高精度水下视觉恢复的研究与应用
水下图像常因散射和吸收等复杂光学效应导致严重色彩失真、可见度低和结构模糊,严重影响视觉质量并制约下游感知任务性能。现有增强方法难以自适应处理多样退化情况,且未能有效利用水下物理先验。本文提出一种退化感知的条件扩散模型,以退化图像为输入,首先通过轻量级双流卷积网络预测其退化水平,生成连续退化分数作为语义引导。基于该分数,设计采用Swin UNet骨干的新型条件扩散恢复网络,实现自适应噪声调度与分层特征优化。为进一步融合水下特定物理先验,引入退化引导的自适应特征融合模块及混合损失函数,包含感知一致性、直方图匹配与特征级对比项。在多个基准数据集上的全面实验表明,本方法能有效恢复水下图像,在颜色保真度、感知质量与结构细节方面表现优异,相较当前最优方法在定量指标与定性评估上均有显著提升。
原文摘要 · Abstract (English)
Underwater images typically suffer from severe colour distortions, low visibility, and reduced structural clarity due to complex optical effects such as scattering and absorption, which greatly degrade their visual quality and limit the performance of downstream visual perception tasks. Existing enhancement methods often struggle to adaptively handle diverse degradation conditions and fail to leverage underwater-specific physical priors effectively. In this paper, we propose a degradation-aware conditional diffusion model to enhance underwater images adaptively and robustly. Given a degraded underwater image as input, we first predict its degradation level using a lightweight dual-stream convolutional network, generating a continuous degradation score as semantic guidance. Based on this score, we introduce a novel conditional diffusion-based restoration network with a Swin UNet backbone, enabling adaptive noise scheduling and hierarchical feature refinement. To incorporate underwater-specific physical priors, we further propose a degradation-guided adaptive feature fusion module and a hybrid loss function that combines perceptual consistency, histogram matching, and feature-level contrast. Comprehensive experiments on benchmark datasets demonstrate that our method effectively restores underwater images with superior colour fidelity, perceptual quality, and structural details. Compared with SOTA approaches, our framework achieves significant improvements in both quantitative metrics and qualitative visual assessments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。