[发明专利]一种基于局部注意力机制的无人车轨迹预测方法

摘要

本发明公开了一种基于局部注意力机制的无人车轨迹预测方法，将无人车周围车的历史轨迹作为输入，充分考虑无人车和邻车间的交互对无人车未来轨迹的影响；根据道路的几何结构和车辆几何外形来构建无人车与邻车的空间交互，再通过局部注意力机制估算与无人车未来轨迹相关度较高的部分车辆，计算这部分车辆与无人车间的相关度，利用相关度加权求和构建时间交互；综合无人车在当前时刻的与周围车的时间、空间交互，将综合交互特征输入后接全连接层的解码器，得到未来一段时间无人车轨迹分布和轨迹坐标；训练时用负对数似然损失函数计算损失，通过损失反向传播更新参数，训练好的模型预测无人车未来一段时间的轨迹，辅助后续决策规划的完成。