arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2008.05927cs.CVcs.RO
收藏

End-to-end Contextual Perception and Prediction with Interaction Transformer

Uber Advanced Technologies Group(优步先进技术集团);University of Toronto(多伦多大学)

Lingyun Luke Li, Bin Yang, Ming Liang, Wenyuan Zeng, Mengye Ren, Sean Segal, Raquel Urtasun

英文摘要:

In this paper, we tackle the problem of detecting objects in 3D and forecasting their future motion in the context of self-driving. Towards this goal, we design a novel approach that explicitly takes into account the interactions between actors. To capture their spatial-temporal dependencies, we propose a recurrent neural network with a novel Transformer architecture, which we call the Interaction Transformer. Importantly, our model can be trained end-to-end, and runs in real-time. We validate our approach on two challenging real-world datasets: ATG4D and nuScenes. We show that our approach can outperform the state-of-the-art on both datasets. In particular, we significantly improve the social compliance between the estimated future trajectories, resulting in far fewer collisions between the predicted actors.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑