GOHOME: Graph-Oriented Heatmap Output for future Motion Estimation - CAO et robotique (CAOR)
Communication Dans Un Congrès Année : 2022

GOHOME: Graph-Oriented Heatmap Output for future Motion Estimation

Résumé

In this paper, we propose GOHOME, a method leveraging graph representations of the High Definition Map and sparse projections to generate a heatmap output representing the future position probability distribution for a given agent in a traffic scene. This heatmap output yields an unconstrained 2D grid representation of agent future possible locations, allowing inherent multimodality and a measure of the uncertainty of the prediction. Our graph-oriented model avoids the high computation burden of representing the surrounding context as squared images and processing it with classical CNNs, but focuses instead only on the most probable lanes where the agent could end up in the immediate future. GOHOME reaches 2nd on Argoverse Motion Forecasting Benchmark on the MissRate6 metric while achieving significant speed-up and memory burden diminution compared to Argoverse 1 st place method HOME. We also highlight that heatmap output enables multimodal ensembling and improve 1 st place MissRate6 by more than 15% with our best ensemble on Argoverse. Finally, we evaluate and reach state-of-the-art performance on the other trajectory prediction datasets nuScenes and Interaction, demonstrating the generalizability of our method.
Fichier principal
Vignette du fichier
GOHOME_ICRA.pdf (1.56 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03683555 , version 1 (31-05-2022)

Identifiants

  • HAL Id : hal-03683555 , version 1

Citer

Thomas Gilles, Stefano Sabatini, Dzmitry Tsishkou, Bogdan Stanciulescu, Fabien Moutarde. GOHOME: Graph-Oriented Heatmap Output for future Motion Estimation. IEEE International Conference on Robotics and Automation (ICRA), May 2022, Philadelphie, United States. ⟨hal-03683555⟩
93 Consultations
253 Téléchargements

Partager

More