基于Swin Transformer改进YOLO的密集场景行人检测算法

张森奕 ,  张雪松 ,  郭佳琦 ,  金花 ,  李光宇

科学技术与工程 ›› 2025, Vol. 25 ›› Issue (21) :9018-9027

PDF (14721KB)
科学技术与工程 ›› 2025, Vol. 25 ›› Issue (21) :9018-9027 DOI: 10.12404/j.issn.1671-1815.2406615
自动化技术、计算机技术

基于Swin Transformer改进YOLO的密集场景行人检测算法

作者信息 +

Improved YOLO Based on Swin Transformer for Dense Scene Pedestrian Detection Algorithm

Author information +
文章历史 +

摘要

在密集场景中,常常包含众多被遮挡或者小尺度的行人目标。这样的场景对常规的目标检测模型提出了挑战,往往会出现大量的漏检和错检问题。为了解决密集场景中行人检测时出现的高漏报率和误报率问题,提出了一种新的密集场景行人检测框架ST-YOLO。首先,将YOLOv5的骨干网络中的低层小目标检测层融入特征金字塔网络和路径聚合网络结构中,增加了一个检测小目标行人检测层;其次,对YOLOv5的颈部网络进行改进,利用基于Swin Transformer的多尺度全局信息和卷积神经网络(convolutional neural networks, CNN)所提取的局部信息来构建聚合特征,提高网络的特征提取能力;并且在预测过程中引入了SIoU(scalable-IoU)损失函数,加快模型的收敛速度和提升检测能力;最后,使用Soft-NMS(soft non-maximum suppression)代替原非极大值抑制(non-maximum suppression, NMS)算法,减少非最大化抑制阶段误删除检测框问题,降低了检测算法的误报率。在Wider Person数据集上的大量实验表明,改进后的ST-YOLO算法的精度和mAP0.5比目前主流的YOLOv9算法分别提升了5.7%和3.6%。

Abstract

In dense scenes, the frequent occurrence of occluded or small-scale pedestrian objects poses significant challenges to traditional object detection models, frequently leading to a high number of missed detections and false positives. In order to solve the problem of high false negative rate and false positive rate in pedestrian detection in such dense scenes, a novel dense scene pedestrian detection framework called ST-YOLO was proposed. Firstly, the low-level small object detection layer in YOLOv5's backbone network was integrated into the feature pyramid network and path aggregation network structure, adding a pedestrian detection layer for detecting small objects. Secondly, the neck network of YOLOv5 was improved by utilizing multi-scale global information based on Swin Transformer and local information extracted by convolutional neural networks (CNN) to construct aggregated features and enhance the network's feature extraction capability. And the SIoU (scalable IoU) loss function was introduced in the prediction process to accelerate the convergence speed of the model and improve detection capability. Finally, Soft NMS (soft non maximum suppression) was used instead of the original non maximum suppression (NMS) algorithm to reduce the problem of mistakenly deleting detection boxes during the non maximum suppression stage and lower the false alarm rate of the detection algorithm. A large number of experiments on the Wide Person dataset have shown that the improved ST-YOLO algorithm has improved accuracy and mAP0.5 by 5.7% and 3.6% respectively compared to the current mainstream YOLOv9 algorithm.

关键词

行人检测 / 密集场景 / Swin Transformer / YOLOv5 / 特征融合

Key words

object detection / dense scenes / Swin Transformer / YOLOv5 / feature fusion

引用本文

引用格式 ▾
张森奕, 张雪松, 郭佳琦, 金花, 李光宇. 基于Swin Transformer改进YOLO的密集场景行人检测算法[J]. 科学技术与工程, 2025, 25(21): 9018-9027 DOI:10.12404/j.issn.1671-1815.2406615
ZHANG Sen-yi, ZHANG Xue-song, GUO Jia-qi, JIN Hua, LI Guang-yu. Improved YOLO Based on Swin Transformer for Dense Scene Pedestrian Detection Algorithm[J]. Science Technology and Engineering, 2025, 25(21): 9018-9027 DOI:10.12404/j.issn.1671-1815.2406615

登录浏览全文

4963

注册一个新账户 忘记密码

参考文献

基金资助

国家自然科学基金(62276042)

辽宁省教育厅科学研究项目(LJKZ0486)

PDF (14721KB)

34

访问

0

被引

详细

导航
相关文章

/

〈 〉