论文标题
基于关键点的全球泳道检测网络
A Keypoint-based Global Association Network for Lane Detection
论文作者
论文摘要
车道检测是一项具有挑战性的任务,需要预测车道线的复杂拓扑形状,并同时区分不同类型的车道。较早的作品遵循自上而下的路线图,将预定义的锚缩回车道线的各种形状,由于固定的锚形状,由于固定的锚固形状而缺乏足够的灵活性来适合复杂的车道形状。最近,一些作品建议将车道检测作为关键点估计问题,以更灵活地描述车道线的形状,并以逐点的方式逐渐将属于同一车道线的相邻关键点组成,这是效率效率且在后加工过程中的时间耗时。在本文中,我们提出了一个全球关联网络(GANET),以从新的角度从新的角度提出车道检测问题,其中每个关键点都直接回归到车道线的起点而不是逐点扩展。具体而言,通过预测其偏移到全球泳道的相应起点而无需彼此依赖的情况下,将关键点与其属于泳道线的关联进行进行,这可以并行地提高效率。此外,我们进一步提出了一条泳道的特征聚合器(LFA),该特征聚合器可自适应地捕获相邻关键点之间的局部相关性,以将本地信息补充到全球关联。在两个流行的车道检测基准上进行的广泛实验表明,我们的方法的表现优于以前的Culane F1得分为79.63%的方法,而Tusimple数据集则比FPS高的Tusimple数据集中的得分为97.71%。该代码将在https://github.com/wolfwjs/ganet上发布。
Lane detection is a challenging task that requires predicting complex topology shapes of lane lines and distinguishing different types of lanes simultaneously. Earlier works follow a top-down roadmap to regress predefined anchors into various shapes of lane lines, which lacks enough flexibility to fit complex shapes of lanes due to the fixed anchor shapes. Lately, some works propose to formulate lane detection as a keypoint estimation problem to describe the shapes of lane lines more flexibly and gradually group adjacent keypoints belonging to the same lane line in a point-by-point manner, which is inefficient and time-consuming during postprocessing. In this paper, we propose a Global Association Network (GANet) to formulate the lane detection problem from a new perspective, where each keypoint is directly regressed to the starting point of the lane line instead of point-by-point extension. Concretely, the association of keypoints to their belonged lane line is conducted by predicting their offsets to the corresponding starting points of lanes globally without dependence on each other, which could be done in parallel to greatly improve efficiency. In addition, we further propose a Lane-aware Feature Aggregator (LFA), which adaptively captures the local correlations between adjacent keypoints to supplement local information to the global association. Extensive experiments on two popular lane detection benchmarks show that our method outperforms previous methods with F1 score of 79.63% on CULane and 97.71% on Tusimple dataset with high FPS. The code will be released at https://github.com/Wolfwjs/GANet.