Autonomous driving requires accurate local scene understanding information. To this end, autonomous agents deploy object detection and online BEV lane graph extraction methods as a part of their perception stack. In this work, we propose an architecture and loss formulation to improve the accuracy of local lane graph estimates by using 3D object detection outputs. The proposed method learns to assign the objects to centerlines by considering the centerlines as cluster centers and the objects as data points to be assigned a probability distribution over the cluster centers. This training scheme ensures direct supervision on the relationship between lanes and objects, thus leading to better performance. The proposed method improves lane graph estimation substantially over state-of-the-art methods. The extensive ablations show that our method can achieve significant performance improvements by using the outputs of existing 3D object detection methods. Since our method uses the detection outputs rather than detection method intermediate representations, a single model of our method can use any detection method at test time.

翻译：自动驾驶需要精确的局部场景理解信息。为此，自动驾驶车辆在其感知栈中部署目标检测与在线BEV车道图提取方法。本文提出一种利用3D目标检测输出结果提升局部车道图估计精度的架构与损失函数设计。该方法通过将中心线视为聚类中心、将目标视为需分配至聚类中心概率分布的数据点，学习将目标与中心线进行关联。这种训练方案确保对车道与目标关系的直接监督，从而获得更优性能。实验表明，所提方法相较现有最优方法显著提升了车道图估计精度。大量消融实验证明，利用现有3D目标检测方法的输出即可实现显著性能提升。由于本方法仅使用检测输出而非检测方法的中间表征，模型在测试阶段可兼容任意检测方法。