Although lane detection methods have shown impressive performance in real-world scenarios, most of methods require post-processing which is not robust enough. Therefore, end-to-end detectors like DEtection TRansformer(DETR) have been introduced in lane detection. However, one-to-one label assignment in DETR can degrade the training efficiency due to label semantic conflicts. Besides, positional query in DETR is unable to provide explicit positional prior, making it difficult to be optimized. In this paper, we present the One-to-Several Transformer(O2SFormer). We first propose the one-to-several label assignment, which combines one-to-one and one-to-many label assignments to improve the training efficiency while keeping end-to-end detection. To overcome the difficulty in optimizing one-to-one assignment. We further propose the layer-wise soft label which adjusts the positive weight of positive lane anchors across different decoder layers. Finally, we design the dynamic anchor-based positional query to explore positional prior by incorporating lane anchors into positional query. Experimental results show that O2SFormer significantly speeds up the convergence of DETR and outperforms Transformer-based and CNN-based detectors on the CULane dataset. Code will be available athttps://github.com/zkyseu/O2SFormer.
翻译:虽然车道检测方法在实际场景中表现出了出色的性能,但是大多数方法都需要后处理,这使得方法不够稳健。因此,端到端检测器如DEtection TRansformer(DETR)被引入到车道检测中。然而,在DETR中,一对一的标签分配可能会由于标签语义冲突而降低训练效率。此外,DETR中的位置查询无法提供明确的位置先验,这使得其难以优化。在本文中,我们提出了一种一对多Transformer(O2SFormer)。我们首先提出了一对多的标签分配方法,将一对一和一对多的标签分配方法相结合,以提高训练效率,同时保持端到端检测。为了克服一对一分配中的优化困难,我们进一步提出了层次软标签,通过调整正向车道锚点在不同解码器层中的正向权重来改善效果。最后,我们设计了基于动态锚点的位置查询方法,将车道锚点用于位置查询,以探索位置先验。实验结果表明,O2SFormer可以显著加速DETR的收敛速度,并在CULane数据集上优于基于Transformer和CNN的检测器。代码将在https://github.com/zkyseu/O2SFormer上公开发布。