三维人体姿态估计模型的训练装置及方法

    公开(公告)号:CN118628858A

    公开(公告)日:2024-09-10

    申请号:CN202310252830.2

    申请日:2023-03-07

    Abstract: 本申请实施例提供一种三维人体姿态估计模型的训练装置及方法。该方法包括:将包括三维图像数据集的训练数据输入到卷积网络中进行特征提取,得到三维热图;根据该三维热图得到人体关节点的三维坐标;根据该人体关节点的三维坐标,计算人体局部方向的预测值、骨骼长度的预测值以及各个人体关节点与根关节点的相对位置的预测值中的至少一个;根据人体局部方向的预测值、骨骼长度的预测值以及各个人体关节点与根关节点的相对位置的预测值中的至少一个与根据该三维图像数据集计算得到的相应真值,确定损失函数;以及基于该损失函数,对三维人体姿态估计模型的参数进行调整。

Patent Agency Ranking