买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！

申请/专利权人：北京旷视科技有限公司;北京迈格威科技有限公司

摘要：本申请提供了一种人体姿态预测模型训练方法、人体姿态预测方法及装置。该方法，包括：获取标注训练集和未标注训练集，标注训练集包括多张包含标注数据的第一人体图像，标注数据用于表示第一人体图像中的真实姿态信息；将第一人体图像输入人体姿态预测模型中的生成器，得到对应的第一人体姿态预测结果，并根据标注数据以及第一人体姿态预测结果计算生成器的第一损失值；将第二人体图像输入生成器，得到对应的第二人体姿态预测结果；根据第一人体图像、标注数据、第二人体图像以及第二人体姿态预测结果计算人体姿态预测模型中的判别器对应的第二损失值；根据第一损失值以及第二损失值对生成器以及判别器进行优化，得到人体姿态预测模型。

主权项：1.一种人体姿态预测模型训练方法，其特征在于，包括：获取标注训练集和未标注训练集，所述标注训练集包括多张包含标注数据的第一人体图像，所述标注数据用于表示所述第一人体图像中的真实姿态信息；所述未标注训练集包括多张不包含标注数据的第二人体图像；将所述第一人体图像输入人体姿态预测模型中的生成器，得到对应的第一人体姿态预测结果，并根据所述标注数据以及所述第一人体姿态预测结果计算所述生成器的第一损失值；将所述第二人体图像输入所述生成器，得到对应的第二人体姿态预测结果；根据所述第一人体图像、所述标注数据、所述第二人体图像以及所述第二人体姿态预测结果计算所述人体姿态预测模型中的判别器对应的第二损失值；根据所述第一损失值以及所述第二损失值对所述生成器以及所述判别器进行优化，得到所述人体姿态预测模型；所述将所述第一人体图像输入人体姿态预测模型中的生成器，得到对应的第一人体姿态预测结果，并根据所述标注数据以及所述第一人体姿态预测结果计算所述生成器的第一损失值，包括：将所述第一人体图像输入所述生成器，得到对应多通道的预测人体姿态热图；其中，每个通道的预测人体姿态热图预测一个人体关键点位置；基于所述第一人体图像对应的标注数据生成对应的参考人体姿态热图；根据所述预测人体姿态热图以及所述参考人体姿态热图计算所述第一损失值；所述根据所述第一人体图像、所述标注数据、所述第二人体图像以及所述第二人体姿态预测结果计算所述人体姿态预测模型中的判别器对应的第二损失值，包括：将所述第一人体图像以及所述参考人体姿态热图作为真数据序列输入所述判别器，并将所述第二人体图像以及所述第二人体姿态预测结果作为假数据序列输入所述判别器，分别得到所述判别器输出的判别结果；根据所述判别结果计算所述第二损失值。

全文数据：

权利要求：

百度查询： 北京旷视科技有限公司 北京迈格威科技有限公司 人体姿态预测模型训练方法、人体姿态预测方法及装置