一种基于多维度自注意力机制的轻量级车道线检测算法

摘要

车道线检测是自动驾驶领域中一个基础的任务。现在最先进的方法主要将车道线检测视为逐像素分割的问题。作为一个密集预测任务,车道线检测需要较大的计算量,因此预测速度较为缓慢。但是作为自动驾驶的子任务,具备实时性是一个重要的要求。所以我们提出了一个极为轻量化的侧导向检测模型,来实现实时的车道线检测。而且针对卷积网络在特征提取过程中所造成的特征之间的依赖减弱问题,我们引入了双维度的自注意力机制来解决这个问题。通过在现有的车道线检测基准数据集上进行的大量实验表明,我们的方法可以在速度和准确性方面达到具备竞争性的性能。