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

通过潜在空间优化实现一致性和多模态图像修复

Coherent and Multi-modality Image Inpainting via Latent Space Optimization

Xi’an Jiaotong University(西安交通大学);EPFL(苏黎世联邦理工学院);University of Chinese Academy of Sciences(中国科学院大学)

Lingzhi Pan, Tong Zhang, Bingyuan Chen, Qi Zhou, Wei Ke, Sabine Süsstrunk, Mathieu Salzmann

AI总结:

PILOT通过潜在空间优化提升图像修复的一致性和多模态处理能力。

摘要:

随着去噪扩散概率模型(DDPMs)的进步,图像修复已从单纯基于邻近区域填充信息发展为根据文本、示例图像和草图等各种提示生成内容。然而,现有方法如模型微调和简单拼接潜在向量往往由于过拟合和修复区域与背景之间的一致性问题导致生成失败。在本文中,我们认为当前大型扩散模型足以在不进一步微调的情况下生成逼真图像。因此,我们引入PILOT(inPainting vIa Latent Optimization),一种基于新颖的语义中心化和背景保留损失的优化方法。我们的方法搜索能够生成与用户提供的提示具有高保真度的修复区域同时与背景保持一致的潜在空间。此外,我们提出了一种平衡优化成本和图像质量的策略,显著提高了生成效率。我们的方法可以无缝集成到任何预训练模型中,包括ControlNet和DreamBooth,使其适用于多模态编辑工具。我们的定性和定量评估表明,PILOT在响应提供的提示时生成的修复区域更加一致、多样和真实。

英文摘要:

With the advancements in denoising diffusion probabilistic models (DDPMs), image inpainting has significantly evolved from merely filling information based on nearby regions to generating content conditioned on various prompts such as text, exemplar images, and sketches. However, existing methods, such as model fine-tuning and simple concatenation of latent vectors, often result in generation failures due to overfitting and inconsistency between the inpainted region and the background. In this paper, we argue that the current large diffusion models are sufficiently powerful to generate realistic images without further tuning. Hence, we introduce PILOT (in\textbf{P}ainting v\textbf{I}a \textbf{L}atent \textbf{O}p\textbf{T}imization), an optimization approach grounded on a novel \textit{semantic centralization} and \textit{background preservation loss}. Our method searches latent spaces capable of generating inpainted regions that exhibit high fidelity to user-provided prompts while maintaining coherence with the background. Furthermore, we propose a strategy to balance optimization expense and image quality, significantly enhancing generation efficiency. Our method seamlessly integrates with any pre-trained model, including ControlNet and DreamBooth, making it suitable for deployment in multi-modal editing tools. Our qualitative and quantitative evaluations demonstrate that PILOT outperforms existing approaches by generating more coherent, diverse, and faithful inpainted regions in response to provided prompts.

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

↑