在计算机视觉领域,车道检测是自动驾驶和智能交通系统中的关键技术之一。我们推出了一种名为LSTR的车道形状预测模型,它基于Transformer架构,实现了端到端的车道形状参数输出。本文将介绍LSTR模型的设计思想、功能特性以及应用场景。
项目背景与意义
传统的车道检测方法通常分为多个阶段,包括特征提取、车道线识别和拟合等,这些阶段需要大量的人工设计和调整,而且容易受到环境光照、路面情况等因素的影响。LSTR的出现将车道检测简化为单阶段任务,极大地提高了检测的准确性和效率。
LSTR模型的特点包括模型参数少、计算复杂度低、训练友好等,不仅在理论上具有吸引力,而且在实际应用中也表现出色。其在TuSimple数据集上达到了96.18%的准确率,为自动驾驶和智能交通领域的应用提供了可靠的技术支持。
LSTR的特性和功能
- 😎 端到端架构: 直接输出车道形状参数,简化车道检测流程。
- ⚡ 超轻量级: 模型参数仅有765,787个,适用于资源受限的环境。
- ⚡ 超低复杂度: 计算复杂度仅为574.280M个乘加操作(MACs)。
- 😎 训练友好: GPU内存消耗低,使用大小为(360, 640, 3)的输入图像,批量大小为16时,仅占用1245MiB的GPU内存。
最新更新
- 【2021/12/03】🔥 我们的新作品《从单张图像学习预测3D车道形状和相机姿态:基于几何约束》已被AAAI2022接受!预印本论文和代码即将发布!
- 【2021/11/23】: 我们现在支持自定义数据的训练和测试。
- 【2021/11/16】: 修复了多GPU训练的问题。
- 【2020/12/06】: 现在我们支持CULane数据集。