马淑康, 蒋华涛, 常琳, 郑琛. 基于注意力机制和特征聚合的车道线检测[J]. 微电子学与计算机, 2022, 39(12): 40-46. DOI: 10.19304/J.ISSN1000-7180.2022.0284
引用本文: 马淑康, 蒋华涛, 常琳, 郑琛. 基于注意力机制和特征聚合的车道线检测[J]. 微电子学与计算机, 2022, 39(12): 40-46. DOI: 10.19304/J.ISSN1000-7180.2022.0284
MA Shukang, JIANG Huatao, CHANG Lin, ZHENG Chen. Lane line detection based on attention mechanism and feature aggregation[J]. Microelectronics & Computer, 2022, 39(12): 40-46. DOI: 10.19304/J.ISSN1000-7180.2022.0284
Citation: MA Shukang, JIANG Huatao, CHANG Lin, ZHENG Chen. Lane line detection based on attention mechanism and feature aggregation[J]. Microelectronics & Computer, 2022, 39(12): 40-46. DOI: 10.19304/J.ISSN1000-7180.2022.0284

基于注意力机制和特征聚合的车道线检测

Lane line detection based on attention mechanism and feature aggregation

  • 摘要: 车道线检测的可靠性和稳定性对智能驾驶系统来说至关重要.由于车道线容易受到光线、遮挡、老化等复杂情况的干扰,导致传统的语义分割网络无法准确的学习到车道线的细节特征.为解决该问题,本文首先在编码网络部分引入CA坐标注意力机制,进一步增强网络对车道线提取能力,然后,在特征聚合网络引入金字塔空洞卷积模块与RESA模块并联来增强模型的感受野,以丰富和提取全局的空间特征信息,最后经过解码网络将融合后的特征图上采样到原图大小,并预测每个车道的位置和概率分布.实验证明,文中提出的算法在CULane数据集上有较高准确率,多路面综合准确率达到76.2%,并通过实车测试表明,该算法检测帧率为30 fps,可以在复杂交通场景下进行实时检测,具有较高的泛化性和鲁棒性.

     

    Abstract: The reliability and stability of lane line detection is very important for intelligent driving system. Because lane lines are easily disturbed by complex situations such as light, occlusion and aging, the traditional semantic segmentation network can not accurately learn the detailed features of lane lines. In order to solve this problem, this paper first introduces the CA coordinate attention mechanism in the coding network to further enhance the ability of the network to extract lane lines, then introduces the pyramid hole convolution module in parallel with the RESA module in the feature aggregation network to enhance the receptive field of the model, so as to enrich and extract the global spatial feature information, and finally samples the fused feature map to the size of the original image through the decoding network, The location and probability distribution of each lane are predicted. Experiments show that the algorithm proposed in this paper has high accuracy on the culane data set, and the comprehensive accuracy of multiple roads reaches 76.2%. The real vehicle test shows that the detection frame rate of the algorithm is 30 fps, which can carry out real-time detection in complex traffic scenes, and has high generalization and robustness.

     

/

返回文章
返回