基于深度学习的轻量化道路交通锥目标检测算法

    Lightweight Road Traffic Cone Detection Algorithm Based on Deep Learning

    • 摘要: 道路交通锥是用于高速公路和城市道路维护的安全隔离标志,采用自动化设备对其进行收放,可大大提高作业效率,并降低传统人工收放方式带来的作业人员安全隐患。对交通锥进行精确的目标检测是确保自动化收放设备正常工作的关键。为提高识别效率,同时保证模型检测准确性并降低模型计算量和内存需求,本文提出了一种轻量化道路交通锥检测的TraCone-YOLO(TraCone-You Only Look Once) 算法。首先,采用基于共享深度可分离卷积的高效检测头,使模型在推理时更加高效;其次,在特征提取过程中使用RepC2f模块,增强模型提取多尺度特征信息的能力;最后,通过引入ELA(Efficient Local Attention) 注意力模块,提出ELA-HSFPN(ELA-Hierarchical Scale-based Feature Pyramid Network) 网络结构进行多尺度特征融合,进一步增强模型的特征表达能力。针对原有的交通锥数据集TraCon缺少不同天气和光照强度的样本,本文对TraCon交通锥数据集进行了数据增强,以模拟实际场景中的不同天气条件,有效扩充训练样本,从而提升模型的泛化能力和鲁棒性。实验结果表明,改进后的模型与原模型相比,参数量和计算量分别减少了56.2%和55.6%,均值平均精度达92.6%,与原模型相比仅下降0.4个百分点。该算法在实现模型轻量化的同时,能够保持良好的检测性能。

       

      Abstract: Road traffic cones are safety isolation signs used for highway and urban road maintenance. Automated equipment can greatly improve operational efficiency and reduce safety hazards for workers caused by traditional manual retrieval methods. Accurate target detection of traffic cones is key to ensuring the normal operation of automated retrieval equipment. To improve recognition efficiency while ensuring model detection accuracy and reducing model computation, a lightweight road traffic cone detection algorithm called TraCone-YOLO (TraCone-You Only Look Once) is proposed in this paper. Firstly, an efficient detection head based on shared depthwise separable convolution is adopted in this paper to make the model more efficient in inference. Secondly, to improve the model's ability to extract multi-scale feature information, the RepC2f module is used in the feature extraction process. Finally, by introducing the ELA (Efficient Local Attention) attention module, an ELA-HSFPN (ELA-Hierarchical Scale based Feature Pyramid Network) network structure is proposed for multi-scale feature fusion, further enhancing the model's feature expression ability.In response to the lack of samples with different weather and light intensities in the original traffic cone dataset called TraCon, data augmentation method is applied to simulate different weather conditions in actual scenarios. The training samples are effectively expanded, thereby improving the model's generalization ability and robustness. The experimental results show that the improved model reduces parameters and computation by 56.2% and 55.6% respectively, with an average precision of 92.6%, which is only 0.4 percentage points lower than the original model. The algorithm proposed in this research can make model lightweight while maintaining good detection performance.

       

    /

    返回文章
    返回