一种解决遮挡的三维多人姿态估计方法及相关装置

本发明公开了一种解决遮挡的三维多人姿态估计方法及相关装置，属于计算机视觉技术领域。该方法通过构建OCC‑MotionBERT模型，首先获取人体姿态数据集并划分为训练集与测试集；将训练数据输入模型进行训练，获得三维人体姿态估计模型并通过测试评估；最终将待测图像或视频输入模型，输出三维关键点坐标。模型采用DSTFormerV2作为特征提取网络，结合图卷积与双流时空注意力机制，有效建模关节间空间关联与时序动态，并引入掩码自编码策略提升对遮挡关键点的推理能力；回归头部分引入不确定性建模，输出三维坐标及置信度。该方法显著增强了在遮挡场景下的姿态补全能力与预测鲁棒性，提高了三维多人姿态估计的精度与实用性。