An Improved YOLOv8 Algorithm for Real-World Road Vehicle Object Detection
Vehicle detection plays a pivotal role in intelligent transportation systems and autonomous driving, ensuring safety, optimizing traffic flow, and supporting advanced driver assistance functions. However, dense vehicle distributions, occlusions, and differing target sizes within complex road environments often challenge the performance of existing detection models. In response, this work introduces an improved vehicle detection framework founded on YOLOv8s. Our design integrates a dual attention mechanism—merging CBAM and SE modules—into the Backbone, thereby reinforcing feature extraction and enhancing detection for smaller targets. Additionally, a cross-layer multi-scale feature fusion strategy, built upon ReP-GFPN, is incorporated into the Neck to boost multi-scale information sharing and strengthen detection across various target dimensions. We further replace the traditional CIOU loss with Wise-IoU, enabling the model to better handle difficult samples and occlusion scenarios. Experiments on the UA-DETRAC dataset demonstrate a 4.8% improvement in [email protected], underscoring the effectiveness of these enhancements in demanding traffic conditions. This work offers a promising avenue for advancing vehicle detection systems in intelligent transportation and autonomous driving contexts.
Authors
- Haoran Ding (ORCID: https://orcid.org/0009-0002-6408-762X)
- Tao Gan (ORCID: https://orcid.org/0000-0002-7331-062X)
- Yuhan Song (ORCID: https://orcid.org/0009-0002-6911-7955)
Institutions
- Wuhan University of Technology (CN)
Publication Details
- Published
- 2024-12-27
- DOI
- https://doi.org/10.1109/auteee62881.2024.10869803
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.00