• 查询稿件
  • 获取最新论文
  • 知晓行业信息

基于改进的YOLOv8多类型工程车检测算法研究

A multi-type engineering vehicle detection algorithm based on improved YOLOv8

  • 摘要: 针对城市轨道交通工地低光照、粉尘等复杂环境下的多重干扰、相似误检及动态漂移等问题,提出一种基于改进的YOLOv8多类型工程车检测算法。该算法在Backbone中建立双模态特征提取体系,用可见光和红外骨干网络提取特征,并利用空间变换网络对齐,通过双通道注意力完成加权融合;Neck引入可变形卷积及关键部件检测头,通过独立回归分支预测部件坐标,并采用L1、GIoU优化;Head利用长短期记忆网络对连续帧进行时序编码,用时空交并比来量化时序一致性及约束检测连续性。实验结果表明,工程车辆识别精确率由91.822%提高到98.625%;召回率从74.777%增加到86.073%;平衡分数由82.121%上升至91.879%。该研究为工地复杂环境中多类别工程车监控提供一种新的方法。

     

    Abstract: Aiming at the problems of multiple interferences, similar false detections and dynamic drift in complex urban rail transit construction site environments with dust and low illumination, this paper proposed a multi-type engineering vehicle detection algorithm based on improved YOLOv8. The algorithm constructed a dual-modal feature extraction system in the Backbone network, extracted features through visible and infrared backbone networks, aligned features via a spatial transformation network, and realized weighted feature fusion through dual-channel attention. In the Neck module, it introduced deformable convolutions and a key component detection head. The network predicted component coordinates through an independent regression branch and adopted L1 loss and Generalized Intersection over Union (GIoU) loss for optimization. In the Head module, it used a Long Short-Term Memory (LSTM) network to perform temporal encoding on continuous video frames, and utilized Spatio-Temporal IoU (ST-IoU) to quantify temporal consistency and constrain detection continuity. Experimental results verified that the precision rate of engineering vehicle recognition increased from 91.822% to 98.625%, the recall rate rose from 74.777% to 86.073%, and the F1-score improved from 82.121% to 91.879%. This study provides a novel method for the monitoring of multi-category engineering vehicles in complex construction site environments.

     

/

返回文章
返回