一种基于注意力机制的人体重建方法、模型、装置

摘要

本发明属于计算机视觉领域，具体涉及一种基于注意力机制的人体重建方法、模型、装置。该重建方法包括如下步骤：步骤一：构建一个人体重建网络模型，所述人体重建网络模型包括特征提取模块、注意力模块、融合模块、参数推断模块和SMPL子模块；步骤二、获取多个包含人物的原始图像，对原始图像进行预处理进而构成训练数据集；步骤三、利用上步骤的训练数据集，通过最小化网络损失函数对所述人体重建网络模型进行训练；步骤四、将待处理的人体图像经过预处理后输入到完成训练的网络模型中，生成具有特定姿态的人体三维模型。本发明解决了现有方法难以根据存在遮挡的单幅人体图像，准确重建出具有准确姿势和形态的三维人体模型的问题。