Research on optimization method of YOLO target detection model integrating Transformer structure
In order to improve the target detection performance of YOLO series models in complex environments, this paper proposes a YOLO optimization method that integrates the Transformer structure. This method introduces Swin Transformer in Backbone to enhance the global modeling capability, introduces SE/CBAM attention mechanism in Neck structure and supplements it with lightweight Transformer module, so as to improve the feature fusion and small target recognition capabilities. The experiment is based on FishEye8K traffic scene dataset, adopts multi-scale training and improved loss function strategy, and combines model pruning and integer quantization to optimize inference efficiency. The results show that the proposed model achieves a good balance between mAP , recall rate and model volume, and the accuracy is improved by 11% compared with YOLOv5, the recall rate is improved by more than 9%, and the inference speed is maintained at 85FPS. Visual detection and false detection analysis verify the effectiveness of the method in multi-target, occluded and distorted environments. This study provides a technical path for the fusion of Transformer and YOLO, and has strong practical deployment value.
Authors
- Jie Jia (ORCID: https://orcid.org/0000-0001-7296-5061)
- Dongming Li (ORCID: https://orcid.org/0000-0002-5531-0618)
- Xinyan Chen (ORCID: https://orcid.org/0000-0002-9174-6558)
- Bo Li (ORCID: https://orcid.org/0000-0003-1109-0163)
Institutions
- Shanghai Dianji University (CN)
Publication Details
- Published
- 2025-06-06
- DOI
- https://doi.org/10.1109/raiie65740.2025.11140069
- Primary Topic
- Advanced Measurement and Detection Methods
- Type
- article
- Field-Weighted Citation Impact
- 0.00