针对自动驾驶中小目标检测难题,提出YOLOv8n-SPTS模型提升精度与鲁棒性。
Traffic Scene Small Target Detection Method Based on YOLOv8n-SPTS Model for Autonomous Driving
- 用空间深度卷积替代传统卷积,保留小目标细粒度特征。
- 引入SPPFCSPC模块增强多尺度特征融合,提升复杂场景理解能力。
- 设计三阶段特征金字塔结构,专用于小目标检测且降低计算开销。
本文聚焦自动驾驶动态感知中的小目标识别难题。现有算法因小目标信息丢失、尺度不平衡和遮挡问题导致检测性能不佳。为此提出改进的YOLOv8n-SPTS模型,通过三项创新:首先,在YOLOv8n的Backbone Bottleneck中用空间到深度卷积(SPD-Conv)替代4个传统卷积模块,通过空间转深度转换保留细粒度信息,减少低分辨率小目标特征损失;其次,引入空间金字塔池化-快速跨阶段部分连接(SPPFCSPC)模块替代原SPPF模块,结合SPP的多尺度提取与CSP的特征融合机制,增强模型对复杂场景的上下文理解与多尺度表达能力;第三,设计三阶段特征金字塔(TSFP)结构,在原有检测头基础上增加160×160分辨率的小目标检测头,充分使用浅层高分辨率特征,并移除冗余的大目标检测头以平衡效率。在VisDrone2019-DET数据集上的对比实验表明,YOLOv8n-SPTS在精度(61.9%)、召回率(48.3%)、[email protected](52.6%)和[email protected]:0.95(32.6%)四项指标上均排名第一。可视化结果验证了在遮挡与密集场景下,行人、自行车等小目标的漏检率显著下降。
原文摘要 · Abstract (English)
This paper focuses on the key issue in autonomous driving: small target recognition in dynamic perception. Existing algorithms suffer from poor detection performance due to missing small target information, scale imbalance, and occlusion. We propose an improved YOLOv8n-SPTS model, which enhances the detection accuracy of small traffic targets through three key innovations: First, optimizing the feature extraction module. In the Backbone Bottleneck structure of YOLOv8n, 4 traditional convolution modules are replaced with Space-to-Depth Convolution (SPD-Conv) modules. This module retains fine-grained information through space-to-depth conversion, reduces information loss, and enhances the ability to capture features of low-resolution small targets. Second, enhancing feature fusion capability. The Spatial Pyramid Pooling - Fast Cross Stage Partial Connection (SPPFCSPC) module is introduced to replace the original SPPF module, integrating the multi-scale feature extraction from Spatial Pyramid Pooling (SPP) and the feature fusion mechanism of Cross Stage Partial Connection (CSP), thereby improving the model's contextual understanding of complex scenes and multi-scale feature expression ability. Third, designing a dedicated detection structure for small targets. A Triple-Stage Feature Pyramid (TSFP) structure is proposed, which adds a 160*160 small target detection head to the original detection heads to fully utilize high-resolution features in shallow layers; meanwhile, redundant large target detection heads are removed to balance computational efficiency. Comparative experiments on the VisDrone2019-DET dataset show that YOLOv8n-SPTS model ranks first in precision (61.9%), recall (48.3%), [email protected] (52.6%), and [email protected]:0.95 (32.6%). Visualization results verify that the miss rate of small targets such as pedestrians and bicycles in occluded and dense scenes is significantly reduced.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。