arXiv:2409.18932cs.CV2024-09被引 6

ReviveDiff用扩散模型统一修复雨天、水下、低光等恶劣天气下的模糊图像。

ReviveDiff: A Universal Diffusion Model for Restoring Images in Adverse Weather Conditions

  • 基于扩散模型,从宏观和微观层面同时优化清晰度、噪声、动态范围等质量因素。
  • 在7个基准数据集上超越现有方法,对雨天、水下、夜间雾霾等5类退化均表现优异。
  • 无需针对每种场景重设计,适合需要通用图像恢复的自动驾驶、安防监控等场景。

在夜间、烟雾、雨天及水下等恶劣环境下拍摄的图像常因严重退化导致视觉质量显著下降,影响后续视觉任务。尽管已有方法针对特定退化引入先验知识并取得成效,但其针对性限制了泛化能力。本文提出一种通用网络架构ReviveDiff,可统一处理多种退化类型,通过增强与恢复图像质量实现“复活”效果。受启发于自然介质(如雾、水、低亮度)引起的退化通常保留物体原始结构这一观察,我们结合扩散模型最新进展,从清晰度、失真、噪声水平、动态范围和色彩准确性等关键维度,在宏观与微观层面协同优化图像质量。我们在涵盖雨天、水下、低光、烟雾和夜间雾霾五类退化条件的七个基准数据集上进行了严格评估,实验结果表明,ReviveDiff在定量指标和视觉效果上均优于现有最先进方法。

原文摘要 · Abstract (English)

Images captured in challenging environments--such as nighttime, smoke, rainy weather, and underwater--often suffer from significant degradation, resulting in a substantial loss of visual quality. The effective restoration of these degraded images is critical for the subsequent vision tasks. While many existing approaches have successfully incorporated specific priors for individual tasks, these tailored solutions limit their applicability to other degradations. In this work, we propose a universal network architecture, dubbed ``ReviveDiff'', which can address various degradations and bring images back to life by enhancing and restoring their quality. Our approach is inspired by the observation that, unlike degradation caused by movement or electronic issues, quality degradation under adverse conditions primarily stems from natural media (such as fog, water, and low luminance), which generally preserves the original structures of objects. To restore the quality of such images, we leveraged the latest advancements in diffusion models and developed ReviveDiff to restore image quality from both macro and micro levels across some key factors determining image quality, such as sharpness, distortion, noise level, dynamic range, and color accuracy. We rigorously evaluated ReviveDiff on seven benchmark datasets covering five types of degrading conditions: Rainy, Underwater, Low-light, Smoke, and Nighttime Hazy. Our experimental results demonstrate that ReviveDiff outperforms the state-of-the-art methods both quantitatively and visually.

图像恢复扩散模型通用修复

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