Mar, 2024

帮助扩散模型:改进条件人类图像生成的两阶段方法

TL;DR我们引入了一种新颖的姿势条件的人体图像生成方法,将生成过程分为两个阶段:手的生成和在手周围绘制身体。我们将手生成器训练在多任务设置中,以产生手图像及其相应的分割掩模,并将训练后的模型用于生成的第一阶段。然后,在第二阶段中使用改进的 ControlNet 模型来绘制生成的手周围的身体,产生最终的结果。我们引入了一种新颖的混合技术,在第二阶段中保留手部细节,以一种连贯的方式结合两个阶段的结果。实验评估证明了我们提出的方法在姿势准确性和图像质量上优于现有技术,在 HaGRID 数据集上得到验证。我们的方法不仅提高了生成手部的质量,还提供了对手势的改进控制,推进了姿势条件的人体图像生成的能力。