复杂交通场景下的目标检测方法

Object Detection Method in Complex Traffic Scenarios

  • 摘要: 针对复杂交通场景目标检测方法的不足,特别是对小目标和遮挡目标的漏检及多尺度目标检测和模型鲁棒性方面的不足,提出了一种改进后的YOLOv8s-SRCEM (You Only Look Once version 8 small model with Small object detection head,Residual Convolutional block attention module,Efficient channel attention module,and Multi-scale block)模型:引入小目标检测头,使模型能够更加敏感地捕捉小尺寸目标,提高对小目标的检测能力;在小目标检测头上集成Res-CBAM (Residual Convolutional Block Attention Module),进一步提高特征学习的显著性;在骨干网络中加入ECA (Efficient Channel Attention)模块,强化模型对特征通道重要性的关注,提升特征选择和模型的鲁棒性;将原始的SPPF (Spatial Pyramid Pooling-Fast)模块替换为MS-Block (Multi-Scale Block),模型在不同尺度上的特征捕捉和融合能力得到增强。在KITTI数据集上,改进后的模型相比于YOLOv8s模型mAP (mean Average Precision)值提高了6.6%。实验结果表明,多种改进方案的组合使模型在复杂交通场景中的检测性能得到全面提升。

     

    Abstract: To address the limitations of target detection methods in complex traffic scenarios, particularly missed detections of small and occluded targets and challenges in multi-scale target detection and model robustness, we propose an improved YOLOv8s-SRCEM model. We introduce a small target detection head to enhance the sensitivity of the model to small-sized targets, improving its detection capability for such objects. Additionally, we integrate the Res-CBAM attention module into the small target detection head to further enhance feature learning salience. We incorporate the ECA module into the backbone network to strengthen the attention to important feature channels of the model and improve feature selection and model robustness. Furthermore, by replacing the original SPPF module with MS-Block, we enhance the feature capture and fusion capabilities of the model across different scales. On the KITTI dataset, the improved model achieves a 6.6% increase in mAP compared to the YOLOv8s model. Experimental results demonstrate that the combination of these enhancements substantially improves the detection performance of the model in complex traffic scenarios.

     

/

返回文章
返回