
科学技术与工程

中国技术经济学会
Improved YOLO Based on Swin Transformer for Dense Scene Pedestrian Detection Algorithm
Sen-yi ZHANG , Xue-song ZHANG , Jia-qi GUO , Hua JIN , Guang-yu LI
Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (21) : 9018 -9027.
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.
object detection / dense scenes / Swin Transformer / YOLOv5 / feature fusion
/
| 〈 |
|
〉 |