arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.08260cs.CVcs.MM

通过语义预修复改进文本引导的目标修复

Improving Text-guided Object Inpainting with Semantic Pre-inpainting

School of Computer Science, Fudan University(复旦大学计算机科学学院);HiDream.ai Inc.(HiDream.ai公司);Shanghai Collaborative Innovation Center of Intelligent Visual Computing(上海智能视觉计算协同创新中心)

Yifu Chen, Jingwen Chen, Yingwei Pan, Yehao Li, Ting Yao, Zhineng Chen, Tao Mei

AI总结:

针对文本引导目标修复中生成质量与可控性不足的问题,提出CAT-Diffusion级联框架,通过语义预修复先推断目标语义特征再引导扩散模型生成,在多个数据集上优于现有方法。

摘要:

近年来,大型文本到图像扩散模型取得了成功,展现出生成高质量图像的显著潜力。对提升图像可编辑性的进一步探索,引发了人们对一项下游任务的浓厚兴趣:根据文本提示描述,在图像指定区域内修复一个全新的目标。然而,该问题并非易事,主要体现在两个方面:1)仅依靠单个U-Net在所有去噪时间步对齐文本提示与视觉目标,不足以生成符合预期的目标;2)在扩散模型复杂的采样空间中,目标生成的可控性无法得到保证。本文提出将典型的单阶段目标修复分解为两个级联过程:1)语义预修复,在多模态特征空间中推断所需目标的语义特征;2)在扩散潜空间中基于这些修复后的语义特征生成高保真度目标。为此,我们将基于Transformer的语义修复器与目标修复扩散模型级联,提出了一种用于文本引导目标修复的新型级联Transformer-扩散(CAscaded Transformer-Diffusion, CAT-Diffusion)框架。在技术层面,语义修复器经过训练,可基于未被掩码的上下文和文本提示预测目标对象的语义特征。语义修复器的输出随后作为信息丰富的视觉提示,通过参考适配器层引导高保真度目标生成,从而实现可控的目标修复。在OpenImages-V6和MSCOCO数据集上的大量评估验证了CAT-Diffusion相较于现有最优方法的优越性。代码可在https://github.com/Nnn-s/CATdiffusion 获取。

英文摘要:

Recent years have witnessed the success of large text-to-image diffusion models and their remarkable potential to generate high-quality images. The further pursuit of enhancing the editability of images has sparked significant interest in the downstream task of inpainting a novel object described by a text prompt within a designated region in the image. Nevertheless, the problem is not trivial from two aspects: 1) Solely relying on one single U-Net to align text prompt and visual object across all the denoising timesteps is insufficient to generate desired objects; 2) The controllability of object generation is not guaranteed in the intricate sampling space of diffusion model. In this paper, we propose to decompose the typical single-stage object inpainting into two cascaded processes: 1) semantic pre-inpainting that infers the semantic features of desired objects in a multi-modal feature space; 2) high-fieldity object generation in diffusion latent space that pivots on such inpainted semantic features. To achieve this, we cascade a Transformer-based semantic inpainter and an object inpainting diffusion model, leading to a novel CAscaded Transformer-Diffusion (CAT-Diffusion) framework for text-guided object inpainting. Technically, the semantic inpainter is trained to predict the semantic features of the target object conditioning on unmasked context and text prompt. The outputs of the semantic inpainter then act as the informative visual prompts to guide high-fieldity object generation through a reference adapter layer, leading to controllable object inpainting. Extensive evaluations on OpenImages-V6 and MSCOCO validate the superiority of CAT-Diffusion against the state-of-the-art methods. Code is available at \url{https://github.com/Nnn-s/CATdiffusion}.

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

↑