3 个月前

Trajectron++:基于异构数据的动态可行性轨迹预测

Trajectron++:基于异构数据的动态可行性轨迹预测

摘要

对人体运动的合理推理是实现安全且具备社会意识的机器人导航的重要前提。因此,多智能体行为预测已成为现代人机交互系统(如自动驾驶汽车)的核心组成部分。尽管已有多种轨迹预测方法,但大多数方法未考虑动态约束,也未能充分融合环境信息(如地图数据)。为此,我们提出Trajectron++,这是一种模块化、图结构的循环模型,能够在融合智能体动力学特性和异构数据(如语义地图)的基础上,对任意数量的多样化智能体进行轨迹预测。Trajectron++设计上可与机器人规划与控制框架紧密集成,例如,其预测结果可选择性地基于自车智能体的运动规划进行条件化。我们在多个具有挑战性的真实世界轨迹预测数据集上验证了该模型的性能,结果表明其在多种先进的确定性与生成式方法中均表现出色,显著优于现有技术水平。

代码仓库

nvlabs/adaptive-prediction
pytorch
GitHub 中提及
StanfordASL/Trajectron-plus-plus
官方
pytorch
GitHub 中提及
nvr-avg/adaptive-prediction
pytorch
GitHub 中提及

基准测试

基准方法指标
trajectory-prediction-on-ethTrajectron++
Avg AMD/AMV 8/12: 1.01
trajectory-prediction-on-ethucyTrajectron++
ADE-8/12: 0.21
FDE-8/12: 0.39
trajectory-prediction-on-nuscenesTrajectron++
MinADE_10: 1.51
MinADE_5: 1.88
MinFDE_1: 9.52
MissRateTopK_2_10: 0.57
MissRateTopK_2_5: 0.7
OffRoadRate: 0.25

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
Trajectron++:基于异构数据的动态可行性轨迹预测 | 论文 | HyperAI超神经