论文标题
多模式轨迹预测的集体意识关注
Class-Aware Attention for Multimodal Trajectory Prediction
论文作者
论文摘要
预测周围动态剂的未来轨迹是自动驾驶中的重要要求。这些轨迹主要取决于周围的静态环境以及这些动态剂的过去运动。此外,代理意图的多模式性质使轨迹预测问题更具挑战性。所有现有模型都同样考虑目标剂以及周围的剂,而无需考虑物理特性的变化。在本文中,我们提出了一个新颖的基于深度学习的框架,用于自主驾驶中的多模式轨迹预测,该框架通过加权注意模块考虑了目标及其周围车辆的物理特性,例如对象类别及其物理尺寸,从而提高了预测的准确性。我们的模型在Nuscenes轨迹预测基准中取得了最高的结果,这些模型是使用栅格图来输入环境信息的模型。此外,我们的模型能够实时运行,实现高度推理率超过300 fps。
Predicting the possible future trajectories of the surrounding dynamic agents is an essential requirement in autonomous driving. These trajectories mainly depend on the surrounding static environment, as well as the past movements of those dynamic agents. Furthermore, the multimodal nature of agent intentions makes the trajectory prediction problem more challenging. All of the existing models consider the target agent as well as the surrounding agents similarly, without considering the variation of physical properties. In this paper, we present a novel deep-learning based framework for multimodal trajectory prediction in autonomous driving, which considers the physical properties of the target and surrounding vehicles such as the object class and their physical dimensions through a weighted attention module, that improves the accuracy of the predictions. Our model has achieved the highest results in the nuScenes trajectory prediction benchmark, out of the models which use rasterized maps to input environment information. Furthermore, our model is able to run in real-time, achieving a high inference rate of over 300 FPS.