用极坐标减少锚点数,实现端到端车道检测
Polar R-CNN: End-to-End Lane Detection with Fewer Anchors
- 结合局部与全局极坐标,灵活生成少量锚点
- 在5个数据集上达到领先性能,且无需非极大值抑制
- 适合追求轻量高效部署的自动驾驶场景
车道检测是自动驾驶中的关键挑战,尤其在真实道路中车道细长、被遮挡时更难识别。现有基于锚点的方法依赖预设锚点提取特征并优化车道位置与形状,但手动设定锚点繁琐,覆盖多样数据需大量密集锚点,且使用非极大值抑制(NMS)去除冗余预测,影响实际部署效率,在复杂场景下表现不佳。本文提出Polar R-CNN,一种端到端的锚点式车道检测方法。通过引入局部与全局极坐标系统,实现灵活锚点生成,显著减少所需锚点数量而性能不降。此外,设计具有启发式结构的三元头,支持无NMS范式,提升密集车道场景下的部署效率与精度。该方法在五个主流数据集(Tusimple、CULane、LLAMAS、CurveLanes、DL-Rai)上取得竞争力结果,兼具轻量化与结构简洁性。源代码已公开于https://github.com/ShqWW/PolarRCNN。
原文摘要 · Abstract (English)
Lane detection is a critical and challenging task in autonomous driving, particularly in real-world scenarios where traffic lanes can be slender, lengthy, and often obscured by other vehicles, complicating detection efforts. Existing anchor-based methods typically rely on prior lane anchors to extract features and subsequently refine the location and shape of lanes. While these methods achieve high performance, manually setting prior anchors is cumbersome, and ensuring sufficient coverage across diverse datasets often requires a large amount of dense anchors. Furthermore, the use of Non-Maximum Suppression (NMS) to eliminate redundant predictions complicates real-world deployment and may underperform in complex scenarios. In this paper, we propose Polar R-CNN, an end-to-end anchor-based method for lane detection. By incorporating both local and global polar coordinate systems, Polar R-CNN facilitates flexible anchor proposals and significantly reduces the number of anchors required without compromising performance.Additionally, we introduce a triplet head with heuristic structure that supports NMS-free paradigm, enhancing deployment efficiency and performance in scenarios with dense lanes.Our method achieves competitive results on five popular lane detection benchmarks--Tusimple, CULane,LLAMAS, CurveLanes, and DL-Rai--while maintaining a lightweight design and straightforward structure. Our source code is available at https://github.com/ShqWW/PolarRCNN.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。