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.

PDF
Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (21) : 9018-9027 DOI: 10.12404/j.issn.1671-1815.2406615
research-article
Improved YOLO Based on Swin Transformer for Dense Scene Pedestrian Detection Algorithm
Author information +
History +
PDF

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.

Keywords

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

Cite this article

Download citation ▾
Sen-yi ZHANG, Xue-song ZHANG, Jia-qi GUO, Hua JIN, Guang-yu LI. Improved YOLO Based on Swin Transformer for Dense Scene Pedestrian Detection Algorithm. Science Technology and Engineering, 2025, 25(21): 9018-9027 DOI:10.12404/j.issn.1671-1815.2406615

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (14721KB)

17

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉