3 个月前

CondLaneNet:一种基于条件卷积的自顶向下车道线检测框架

CondLaneNet:一种基于条件卷积的自顶向下车道线检测框架

摘要

基于深度学习的现代车道线检测方法在多数场景下表现优异,但在处理具有复杂拓扑结构的车道线时仍面临挑战。本文提出了一种新型的自顶向下式车道线检测框架——CondLaneNet,该方法首先检测车道实例,随后动态预测每个实例的线形轮廓。为解决车道实例级别的区分难题,我们引入了一种基于条件卷积与逐行建模的条件化车道线检测策略。此外,我们设计了循环实例模块(Recurrent Instance Module, RIM),以有效应对密集车道线、分叉车道线等复杂拓扑结构的检测问题。得益于端到端的处理流程,该方法几乎无需后处理,因而具备实时性优势。我们在三个主流车道线检测基准数据集上进行了广泛实验,结果表明,所提方法在全部三个数据集上均达到了当前最优性能。尤其值得一提的是,该方法在准确率与效率之间实现了良好平衡:在CULane数据集上,取得了78.14的F1分数与220 FPS的推理速度。相关代码已开源,地址为:https://github.com/aliyun/conditional-lane-detection。

代码仓库

基准测试

基准方法指标
lane-detection-on-culaneCondLaneNet-L(ResNet-101)
F1 score: 79.48
lane-detection-on-culaneCondLaneNet-M(ResNet-34)
F1 score: 78.74
lane-detection-on-culaneCondLaneNet-S(ResNet-18)
F1 score: 78.14
lane-detection-on-curvelanesCondLaneNet-L(ResNet-101)
F1 score: 86.10
FPS: 48
GFLOPs: 44.9
Precision: 88.98
Recall: 83.41
lane-detection-on-curvelanesCondLaneNet-M(ResNet-34)
F1 score: 85.92
FPS: 109
GFLOPs: 19.7
Precision: 88.29
Recall: 83.68
lane-detection-on-curvelanesCondLaneNet-S(ResNet-18)
F1 score: 85.09
FPS: 154
GFLOPs: 10.3
Precision: 87.75
Recall: 82.58
lane-detection-on-tusimpleCondLaneNet-M(ResNet-34)
Accuracy: 95.37%
F1 score: 96.98
lane-detection-on-tusimpleCondLaneNet(ResNet-18)
Accuracy: 95.48%
lane-detection-on-tusimpleCondLaneNet(ResNet-34)
F1 score: 97.01
lane-detection-on-tusimpleCondLaneNet-L(ResNet-101)
Accuracy: 96.54%
F1 score: 97.24

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
CondLaneNet:一种基于条件卷积的自顶向下车道线检测框架 | 论文 | HyperAI超神经