3 个月前

基于逐行分类的端到端车道线检测

基于逐行分类的端到端车道线检测

摘要

在自动驾驶领域,准确且可靠地检测车道线位置是一项至关重要但极具挑战性的任务。传统的车道线检测方法通常采用像素级密集预测,随后依赖复杂的后处理步骤——这一过程不可避免,因为车道线通常以无厚度的线段集合形式表示。本文提出一种端到端的直接车道线顶点预测方法,无需传统密集预测中必需的后处理步骤。具体而言,我们将车道线检测问题转化为逐行分类任务,充分利用了车道线固有的几何形态特征,而这一思路在以往研究中尚未得到充分探索。为高效提取图像中从左至右延伸的车道线信息,我们设计了一种新型网络层,能够逐级压缩水平方向的特征分量,从而构建一个端到端的车道线检测系统,测试阶段仅通过argmax操作即可直接获得最终的车道线位置。实验结果表明,所提方法在两个主流车道线检测基准数据集(TuSimple 和 CULane)上性能达到或超过当前最优方法,充分验证了其有效性。

代码仓库

基准测试

基准方法指标
lane-detection-on-culaneERFNet-E2E
F1 score: 74
lane-detection-on-culaneResNet-101-E2E
F1 score: 71.9
lane-detection-on-tusimpleR-34-E2E
Accuracy: 96.22%
F1 score: 96.58
lane-detection-on-tusimpleR-50-E2E
Accuracy: 96.11%
F1 score: 96.37
lane-detection-on-tusimpleERF-E2E
Accuracy: 96.02%
F1 score: 96.25

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于逐行分类的端到端车道线检测 | 论文 | HyperAI超神经