3 个月前

CLRerNet:基于LaneIoU提升车道检测置信度

CLRerNet:基于LaneIoU提升车道检测置信度

摘要

车道线检测是自动驾驶与驾驶员辅助系统中的关键组成部分。现有的基于行(row-based)表示的深度车道检测方法在车道检测基准测试中表现出色。通过初步的Oracle实验,我们首先解耦了车道表示的各个组成部分,从而明确了研究方向。实验结果表明,现有基于行的检测器的预测中已包含正确的车道位置,而能够准确反映预测结果与真实标注之间交并比(IoU)的置信度分数最具价值。基于这一发现,我们提出了LaneIoU方法,该方法通过引入局部车道角度信息,使置信度得分与评估指标的相关性进一步增强。在此基础上,我们设计了一种新型检测器——CLRerNet,其在目标分配代价函数与损失函数中引入LaneIoU,旨在提升置信度分数的质量。通过精心设计且公平的基准测试(包括交叉验证),我们证明CLRerNet显著优于现有最先进方法:在CULane数据集上,F1得分达到81.43%,优于现有方法的80.47%;在CurveLanes数据集上,F1得分达到86.47%,相较现有方法的86.10%亦有明显提升。

代码仓库

hirotomusiker/clrernet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
lane-detection-on-culaneCLRerNet-DLA34
F1 score: 81.12
lane-detection-on-culaneCLRerNet-Res101
F1 score: 80.91
lane-detection-on-culaneCLRerNet-Res34
F1 score: 80.76
lane-detection-on-curvelanesCLRNet-DLA34
F1 score: 86.1
GFLOPs: 18.4
Precision: 91.4
Recall: 81.39
lane-detection-on-curvelanesCLRerNet-DLA34
F1 score: 86.47
GFLOPs: 18.4
Precision: 91.66
Recall: 81.83

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
CLRerNet:基于LaneIoU提升车道检测置信度 | 论文 | HyperAI超神经