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

Stable-Pose:利用Transformer进行姿态引导的文本到图像生成

Stable-Pose: Leveraging Transformers for Pose-Guided Text-to-Image Generation

Technical University of Munich (TUM)(慕尼黑工业大学);Munich Center for Machine Learning (MCML)(慕尼黑机器学习中心);LMU Munich(慕尼黑大学)

Jiajun Wang, Morteza Ghahremani, Yitong Li, Björn Ommer, Christian Wachinger

AI总结:

针对复杂人体姿态下文本到图像生成精度不足的问题,提出Stable-Pose适配器,利用ViT的粗到细注意力掩蔽策略,在Stable Diffusion中实现精确姿态引导,并在LAION-Human上AP达57.1,较ControlNet提升约13%。

摘要:

可控文本到图像(T2I)扩散模型在通过整合各种条件生成高质量视觉内容方面展现了令人印象深刻的表现。然而,当前方法在受骨架人体姿态引导时表现有限,尤其是在复杂姿态条件下,如人体的侧面或背面视角。为解决此问题,我们提出了Stable-Pose,一种新颖的适配器模型,它将从粗到细的注意力掩蔽策略引入视觉Transformer(ViT),以获取T2I模型的精确姿态引导。Stable-Pose旨在熟练处理预训练Stable Diffusion中的姿态条件,在图像合成过程中提供一种精细且高效的姿态表示对齐方式。我们利用ViT的查询-键自注意力机制来探索人体姿态骨架中不同解剖部位之间的相互联系。掩蔽姿态图像用于基于目标姿态相关特征,以从粗到细的层次方式平滑地细化注意力图。此外,我们的损失函数被设计为对姿态区域赋予更多重视,从而增强模型捕捉复杂姿态细节的精度。我们在五个公共数据集上,在广泛的室内和室外人体姿态场景下评估了Stable-Pose的性能。Stable-Pose在LAION-Human数据集上取得了57.1的AP分数,相比已有技术ControlNet提升了约13%。项目链接和代码可在https://github.com/ai-med/StablePose获取。

英文摘要:

Controllable text-to-image (T2I) diffusion models have shown impressive performance in generating high-quality visual content through the incorporation of various conditions. Current methods, however, exhibit limited performance when guided by skeleton human poses, especially in complex pose conditions such as side or rear perspectives of human figures. To address this issue, we present Stable-Pose, a novel adapter model that introduces a coarse-to-fine attention masking strategy into a vision Transformer (ViT) to gain accurate pose guidance for T2I models. Stable-Pose is designed to adeptly handle pose conditions within pre-trained Stable Diffusion, providing a refined and efficient way of aligning pose representation during image synthesis. We leverage the query-key self-attention mechanism of ViTs to explore the interconnections among different anatomical parts in human pose skeletons. Masked pose images are used to smoothly refine the attention maps based on target pose-related features in a hierarchical manner, transitioning from coarse to fine levels. Additionally, our loss function is formulated to allocate increased emphasis to the pose region, thereby augmenting the model's precision in capturing intricate pose details. We assessed the performance of Stable-Pose across five public datasets under a wide range of indoor and outdoor human pose scenarios. Stable-Pose achieved an AP score of 57.1 in the LAION-Human dataset, marking around 13% improvement over the established technique ControlNet. The project link and code is available at https://github.com/ai-med/StablePose.

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

↑