3 个月前

基于网格规划的未知环境轨迹预测

基于网格规划的未知环境轨迹预测

摘要

我们针对在未知环境中基于历史运动状态与场景结构预测行人及车辆轨迹的问题展开研究。由于场景结构差异显著以及未来轨迹呈现多模态分布,轨迹预测是一项极具挑战性的任务。与以往直接从观测上下文学习“单对多”映射关系以生成多个未来轨迹的方法不同,本文提出一种新方法:将轨迹预测条件化于通过最大熵逆强化学习(MaxEnt IRL)所学习的基于网格的策略所采样的规划(plans)。我们对MaxEnt IRL进行重构,使该策略能够在场景上定义的粗粒度二维网格中,联合推断出合理的智能体目标及其通往目标的路径。在此基础上,我们设计了一种基于注意力机制的轨迹生成器,能够根据从MaxEnt策略中采样的状态序列,生成连续值的未来轨迹。在公开可用的Stanford Drone与NuScenes数据集上的定量与定性评估表明,所提模型生成的轨迹具有高度多样性,能够有效捕捉多模态预测分布,同时在较长预测时域内仍能精确符合场景结构的物理约束。

代码仓库

nachiket92/P2T
官方
pytorch

基准测试

基准方法指标
trajectory-prediction-on-nuscenesP2T
MinADE_10: 1.16
MinADE_5: 1.45
MinFDE_1: 10.5
MissRateTopK_2_10: 0.46
MissRateTopK_2_5: 0.64
OffRoadRate: 0.03
trajectory-prediction-on-stanford-droneP2TIRL
ADE-8/12 @K = 20: 12.58
FDE-8/12 @K= 20: 22.07

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于网格规划的未知环境轨迹预测 | 论文 | HyperAI超神经