基于多尺度特征融合的机动车黑烟检测

VEHICLE BLACK SMOKE DETECTION ALGORITHM BASED ON MULTI-SCALE FEATURE FUSION

  • 摘要: 针对交通场景下机动车黑烟检测存在的漏检、误检问题,提出一种基于多尺度特征融合的机动车黑烟目标检测算法。在YOLOv5的基础上,重构多尺度网络模型,增加一个小目标层用于特征融合与检测,来提升网络对小目标黑烟的响应能力。同时,在进行多尺度特征融合时,将路径聚合网络PANet替换为双向加权特征金字塔网络BiFPN,融合更多的黑烟特征信息并调整不同尺度特征在网络中的贡献度。实验结果表明,提出的YOLOv5s-FB方法黑烟检测率和非黑烟检测率分别为92.34%和91.75%,检测速度可达35.3FPS,能够满足实际应用需求。

     

    Abstract: Aimed at the problems of missed detection and false detection in vehicle black smoke detection in traffic scenarios, a vehicle black smoke target detection algorithm based on multi-scale feature fusion is proposed. On the basis of YOLOv5, we reconstructed the multi-scale network, added a small target layer for feature fusion and detection, so that the network’s response ability to small target black smoke was improved. At the same time, when performing multi-scale feature fusion, the path aggregation network (PANet) was replaced by the bidirectional feature pyramid network (BiFPN), which integrated more smoky feature information and adjusted the contribution of smoky features at different scales in the network. The experimental results show that the black smoke detection rate and non-black smoke detection rate of the proposed YOLOv5s-FB method are 92.34% and 91.75%, respectively, and the detection speed can reach 35.3FPS, which can meet the actual application requirements.

     

/

返回文章
返回