龚志力, 谷玉海, 朱腾腾, 石文天. 融合注意力机制与轻量化DeepLabv3+的非结构化道路识别[J]. 微电子学与计算机, 2022, 39(2): 26-33. DOI: 10.19304/J.ISSN1000-7180.2021.0704
引用本文: 龚志力, 谷玉海, 朱腾腾, 石文天. 融合注意力机制与轻量化DeepLabv3+的非结构化道路识别[J]. 微电子学与计算机, 2022, 39(2): 26-33. DOI: 10.19304/J.ISSN1000-7180.2021.0704
GONG Zhili, GU Yuhai, ZHU Tengteng, SHI Wentian. Unstructured road recognition based on attention mechanism and lightweight DeepLabv3+[J]. Microelectronics & Computer, 2022, 39(2): 26-33. DOI: 10.19304/J.ISSN1000-7180.2021.0704
Citation: GONG Zhili, GU Yuhai, ZHU Tengteng, SHI Wentian. Unstructured road recognition based on attention mechanism and lightweight DeepLabv3+[J]. Microelectronics & Computer, 2022, 39(2): 26-33. DOI: 10.19304/J.ISSN1000-7180.2021.0704

融合注意力机制与轻量化DeepLabv3+的非结构化道路识别

Unstructured road recognition based on attention mechanism and lightweight DeepLabv3+

  • 摘要: 由于非结构化道路特征众多、结构复杂的特点,图像分割以及道路模型等经典算法无法满足非结构化道路识别在实际应用中的准确性和实时性要求.上述难点可通过基于深度学习的语义分割算法有效解决,采用轻量化的特征提取网络,改善特征提取网络中离散计算过多问题,优化对参数量和速度的控制,减少DeepLabv3+网络的冗余;针对非结构化道路在图像中空间和通道的分布特性,引入注意力机制对高级特征图和低级特征图处理,以提升网络对特征提取的敏感性和准确性;选用样本类别充足并具代表性且与我国路况相似性强的IDD(India Driving Dataset)非结构化道路数据集,对其中非结构化道路类别进行筛选和预处理并对网络进行训练.实验对比表明,相比于同类网络,参数量最少,准确率和快速性均处于前列;相比于改进前,网络参数量减少93.1%,识别帧率提升116.1%,精确率提高10.1%.

     

    Abstract: Due to the numerous features and complex structure of unstructured road, classical algorithms such as image segmentation and road model cannot meet the accuracy and real-time requirements of unstructured road identification in practical application. The above difficulties can be effectively solved by the semantic segmentation algorithm based on deep learning. The lightweight feature extraction network is adopted to improve the excessive discrete computation in the feature extraction network, optimize the control of the number of parameters and speed, and greatly reduce the redundancy of Deeplabv3+ network. According to the spatial and channel distribution characteristics of unstructured roads in the image, attention mechanism is introduced to process the high-level and low-level feature images to improve the sensitivity and accuracy of the network to feature extraction. IDD (India Driving Dataset) unstructured road data sets with sufficient and representative sample categories and strong similarity to China′s road conditions were selected to screen and preprocess the unstructured road categories and train the network. Experimental comparison shows that compared with similar networks, the number of parameters is the least, and the accuracy and rapidity are in the forefront. Compared with before the improvement, the number of network parameters decreases by 93.1%, the recognition frame rate increases by 116.1%, and the accuracy rate increases by 10.1%.

     

/

返回文章
返回