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

Institutions

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
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Research on optimization method of YOLO target detection model integrating Transformer structure

Jie Jia, Dongming Li, Xinyan Chen, Bo Li
Advanced Measurement and Detection Methods
article

Research on optimization method of YOLO target detection model integrating Transformer structure

Jie Jia, Dongming Li, Xinyan Chen, Bo Li
article en

Abstract

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.

Shanghai Dianji University (CN)
Openalex Percentile: Top 78%
Advanced Measurement and Detection Methods
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.