本申请公开了一种基于时空注意力的轨迹预测方法、装置、设备及介质,涉及轨迹预测领域,该方法包括根据目标场景中的目标构建当前时刻的时空图;将当前时刻的时空图输入轨迹预测模型中,得到目标场景中的目标在下一时刻的位置;其中,轨迹预测模型是基于时空注意力机制循环神经网络构建的;时空注意力机制循环神经网络包括:边缘编码层和节点状态转换层;边缘编码层是基于循环神经网络构建的;状态节点转换层是基于多头注意力机制和门控循环单元构建的,本申请能提高目标的轨迹预测精度。
背景技术
近年来,时空预测系统受到了越来越多的关注,特别是在自动驾驶、智能监控、行人轨迹预测等领域。随着各类应用场景的复杂化和智能化,如何准确地预测目标的未来轨迹已成为一个重要的研究课题。在这些场景中,目标的运动轨迹受到多种因素的影响,包括历史轨迹、空间位置的动态变化、周围环境的干扰等。
相关的轨迹预测方法在应对高密度场景、多目标动态交互时,仍然存在精度不高、计算效率低下等问题。例如,在行人密集的环境中,不同目标之间的相互影响复杂多变,相关的轨迹预测方法难以捕捉这种复杂的空间依赖关系,导致预测误差较大。同时,随着预测时间步的增加,预测精度逐渐下降,模型无法有效处理长时间的轨迹预测任务。
实现思路