一种基于多层次时空图神经网络的人群轨迹预测方法

摘要

本发明提供了一种基于多层次时空图神经网络的人群轨迹预测方法，属于计算机视觉和人群轨迹预测技术领域。本发明方法包括：从场景中各行人的历史轨迹中提取行人的运动状态特征；将行人运动过程建模为一个多层次动态时空有向图，使用一个多层次时空图神经网络进行处理，提取得到每个行人的运动交互特征，在提取时，在不同层次分别提取不同距离范围的交互特征，再将不同层次的交互特征融合得到最终的交互特征；基于运动交互特征和运动状态特征预测行人的未来轨迹。本发明通过将不同距离范围的行人交互建模在图的不同层次来差异化建模不同距离范围相同类型交互的影响，使行人交互建模更细致合理，从而提升人群轨迹预测精度。