arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2207.14626cs.CVcs.LG

使用基于 Patch 的去噪扩散模型在恶劣天气条件下恢复视觉

Restoring Vision in Adverse Weather Conditions with Patch-Based Denoising Diffusion Models

Graz University of Technology(格拉茨工业大学);Silicon Austria Labs(奥地利硅实验室)

Ozan Özdenizci, Robert Legenstein

AI总结:

本文提出基于 patch 的去噪扩散概率模型,通过重叠 patch 的平滑噪声估计与引导去噪实现任意尺寸图像恢复,并在多类恶劣天气任务中达到最先进性能。

摘要:

恶劣天气条件下的图像恢复在各类计算机视觉应用中一直受到广泛关注。近期成功的方法依赖于深度神经网络架构设计的当前进展(例如 vision transformer)。受最先进条件生成模型所取得最新进展的启发,我们提出了一种基于去噪扩散概率模型的新型基于 patch 的图像恢复算法。我们的基于 patch 的扩散建模方法通过在推理期间使用跨重叠 patch 的平滑噪声估计的引导去噪过程,实现了与尺寸无关的图像恢复。我们在用于图像去雪、联合去雨和去雾以及雨滴去除的基准数据集上对模型进行了实证评估。我们证明了该方法在特定天气和多天气图像恢复上均取得最先进性能,并通过实验展示了对真实世界测试图像的强泛化能力。

英文摘要:

Image restoration under adverse weather conditions has been of significant interest for various computer vision applications. Recent successful methods rely on the current progress in deep neural network architectural designs (e.g., with vision transformers). Motivated by the recent progress achieved with state-of-the-art conditional generative models, we present a novel patch-based image restoration algorithm based on denoising diffusion probabilistic models. Our patch-based diffusion modeling approach enables size-agnostic image restoration by using a guided denoising process with smoothed noise estimates across overlapping patches during inference. We empirically evaluate our model on benchmark datasets for image desnowing, combined deraining and dehazing, and raindrop removal. We demonstrate our approach to achieve state-of-the-art performances on both weather-specific and multi-weather image restoration, and experimentally show strong generalization to real-world test images.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑