论文标题
TPNET:运动预测的轨迹提案网络
TPNet: Trajectory Proposal Network for Motion Prediction
论文作者
论文摘要
对周围交通代理(例如行人,车辆和骑自行车者)进行准确的运动预测对于自动驾驶至关重要。最近,数据驱动的运动预测方法试图学会直接从大量轨迹数据中回归确切的未来位置或其分布。但是,这些方法仍然很难提供多模式预测以及整合物理约束,例如交通规则和可移动区域。在这项工作中,我们提出了一个新型的两阶段运动预测框架,轨迹提案网络(TPNET)。 TPNET首先生成一组未来的轨迹作为假设建议,然后通过对符合物理约束的建议进行分类和完善提案来做出最终预测。通过指导提案生成过程,可以实现安全和多模式的预测。因此,该框架有效地减轻了运动预测问题的复杂性,同时确保多模式输出。在四个大规模轨迹预测数据集(即ETH,UCY,APOLLO和ARGOVORSE数据集)上进行了实验,表明TPNET在数量和定性上都实现了最先进的结果。
Making accurate motion prediction of the surrounding traffic agents such as pedestrians, vehicles, and cyclists is crucial for autonomous driving. Recent data-driven motion prediction methods have attempted to learn to directly regress the exact future position or its distribution from massive amount of trajectory data. However, it remains difficult for these methods to provide multimodal predictions as well as integrate physical constraints such as traffic rules and movable areas. In this work we propose a novel two-stage motion prediction framework, Trajectory Proposal Network (TPNet). TPNet first generates a candidate set of future trajectories as hypothesis proposals, then makes the final predictions by classifying and refining the proposals which meets the physical constraints. By steering the proposal generation process, safe and multimodal predictions are realized. Thus this framework effectively mitigates the complexity of motion prediction problem while ensuring the multimodal output. Experiments on four large-scale trajectory prediction datasets, i.e. the ETH, UCY, Apollo and Argoverse datasets, show that TPNet achieves the state-of-the-art results both quantitatively and qualitatively.