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

DiffBIR:利用生成式扩散先验实现盲图像复原

DiffBIR: Towards Blind Image Restoration with Generative Diffusion Prior

Shenzhen Institute of Advanced Technology, Chinese Academy of Sciences(中国科学院深圳先进技术研究院);Shanghai AI Laboratory(上海人工智能实验室);The Chinese University of Hong Kong(香港中文大学)

Xinqi Lin, Jingwen He, Ziyan Chen, Zhaoyang Lyu, Bo Dai, Fanghua Yu, Wanli Ouyang, Yu Qiao, Chao Dong

AI总结:

本研究提出通用盲图像复原框架DiffBIR,将任务解耦为退化去除与信息再生两阶段,结合IRControlNet和区域自适应引导,在多项盲复原任务上性能超越现有最优方法。

摘要:

我们提出了DiffBIR,这是一种通用的复原流程,能够在统一框架下处理不同的盲图像复原任务。DiffBIR将盲图像复原问题解耦为两个阶段:1)退化去除:移除与图像内容无关的成分;2)信息再生:生成丢失的图像内容。两个阶段独立开发,但能以级联方式无缝协作。在第一阶段,我们使用复原模块去除退化,得到高保真的复原结果。针对第二阶段,我们提出了IRControlNet,它利用潜在扩散模型的生成能力生成逼真的细节。具体而言,IRControlNet基于专门制作的、不含干扰性噪声内容的条件图像进行训练,以实现稳定的生成性能。此外,我们设计了一种区域自适应复原引导机制,可在推理过程中修改去噪过程而无需重新训练模型,用户能够通过可调节的引导尺度在真实感和保真度之间取得平衡。大量实验表明,在合成数据集和真实世界数据集上,DiffBIR在盲图像超分辨率、盲人脸复原和盲图像去噪任务上均优于现有最优方法。代码可在https://github.com/XPixelGroup/DiffBIR 获取。

英文摘要:

We present DiffBIR, a general restoration pipeline that could handle different blind image restoration tasks in a unified framework. DiffBIR decouples blind image restoration problem into two stages: 1) degradation removal: removing image-independent content; 2) information regeneration: generating the lost image content. Each stage is developed independently but they work seamlessly in a cascaded manner. In the first stage, we use restoration modules to remove degradations and obtain high-fidelity restored results. For the second stage, we propose IRControlNet that leverages the generative ability of latent diffusion models to generate realistic details. Specifically, IRControlNet is trained based on specially produced condition images without distracting noisy content for stable generation performance. Moreover, we design a region-adaptive restoration guidance that can modify the denoising process during inference without model re-training, allowing users to balance realness and fidelity through a tunable guidance scale. Extensive experiments have demonstrated DiffBIR's superiority over state-of-the-art approaches for blind image super-resolution, blind face restoration and blind image denoising tasks on both synthetic and real-world datasets. The code is available at https://github.com/XPixelGroup/DiffBIR.

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

↑