一种基于时空注意力的轨迹预测方法、装置、设备及介质

本申请公开了一种基于时空注意力的轨迹预测方法、装置、设备及介质，涉及轨迹预测领域，该方法包括根据目标场景中的目标构建当前时刻的时空图；将当前时刻的时空图输入轨迹预测模型中，得到目标场景中的目标在下一时刻的位置；其中，轨迹预测模型是基于时空注意力机制循环神经网络构建的；时空注意力机制循环神经网络包括：边缘编码层和节点状态转换层；边缘编码层是基于循环神经网络构建的；状态节点转换层是基于多头注意力机制和门控循环单元构建的，本申请能提高目标的轨迹预测精度。