arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.03490cs.CV

SceneBooth:用于主体保持的文本到图像生成的基于扩散的框架

SceneBooth: Diffusion-based Framework for Subject-preserved Text-to-Image Generation

University of Science and Technology of China(中国科学技术大学);Alibaba Group(阿里巴巴集团)

Shang Chai, Zihang Lin, Min Zhou, Xubin Li, Liansheng Zhuang, Houqiang Li

AI总结:

本文提出 SceneBooth 框架,通过固定主体图像并生成协调背景,结合多模态布局与 ControlNet、门控自注意力适配器,提升文本到图像生成中的主体保真度。

摘要:

由于个性化图像生成的需求,主体驱动的文本到图像生成方法(即根据文本提示创建输入主体的新颖再现)受到了越来越多的研究关注。现有方法通常学习主体表示并将其融入提示嵌入以引导图像生成,但它们在保持主体保真度方面存在困难。为解决这一问题,本文提出了一个名为 SceneBooth 的新颖框架,用于主体保持的文本到图像生成,该框架接收主体图像、对象短语和文本提示作为输入。与学习主体表示并生成主体不同,我们的 SceneBooth 固定给定的主体图像,并在文本提示引导下生成其背景图像。为此,SceneBooth 引入了两个关键组件,即多模态布局生成模块和背景绘制模块。前者通过生成与文本标题、对象短语和主体视觉信息对齐的适当场景布局,确定主体的位置和尺度。后者将两个适配器(ControlNet 和门控自注意力)集成到潜在扩散模型中,以在场景布局和文本描述引导下生成与主体协调的背景。通过这种方式,SceneBooth 确保在输出中准确保持主体的外观。定量和定性实验结果表明,SceneBooth 在主体保持、图像协调和整体质量方面显著优于基线方法。

英文摘要:

Due to the demand for personalizing image generation, subject-driven text-to-image generation method, which creates novel renditions of an input subject based on text prompts, has received growing research interest. Existing methods often learn subject representation and incorporate it into the prompt embedding to guide image generation, but they struggle with preserving subject fidelity. To solve this issue, this paper approaches a novel framework named SceneBooth for subject-preserved text-to-image generation, which consumes inputs of a subject image, object phrases and text prompts. Instead of learning the subject representation and generating a subject, our SceneBooth fixes the given subject image and generates its background image guided by the text prompts. To this end, our SceneBooth introduces two key components, i.e., a multimodal layout generation module and a background painting module. The former determines the position and scale of the subject by generating appropriate scene layouts that align with text captions, object phrases, and subject visual information. The latter integrates two adapters (ControlNet and Gated Self-Attention) into the latent diffusion model to generate a background that harmonizes with the subject guided by scene layouts and text descriptions. In this manner, our SceneBooth ensures accurate preservation of the subject's appearance in the output. Quantitative and qualitative experimental results demonstrate that SceneBooth significantly outperforms baseline methods in terms of subject preservation, image harmonization and overall quality.

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

↑