A Lightweight Object Detection Network for Vehicle Environment Perception Based on Improved YOLOv5s
Vehicle environment perception techniques are crucial for ensuring safe driving conditions on roadways. Vision-based systems can achieve high accuracy and real-time processing in detecting and recognizing objects, which makes them versatile across different vehicle types and driving conditions, enhancing their suitability for various applications in intelligent transportation systems. However, existing object detection methods often suffer from a high number of parameters and significant computational demands, which limit their performance when deployed on vehicle hardware. To tackle these challenges, a lightweight target detection network based on YOLOv5s framework is designed for vehicle environment perception. First, the proposed network integrates depthwise separable convolution to reduce network complexity and redesigns the C3 module based on one-shot aggregation to streamline the overall network structure. Then, CA mechanism is also introduced to learn each channel feature adaptively and focuse on key feature areas to improve the detection performance of the network. Third, MPDloU is employed as a bounding box loss function to improve the robust and reliable object detection performance, particularly in challenging scenarios with varied object sizes and aspect ratios. Experimental results on the KITTI public dataset demonstrate that the improved model achieves 34.7%, 35.4% and 33.6% respectively in parameters, GFLOPs and model size.
Authors
- Ge Ma (ORCID: https://orcid.org/0000-0002-6025-864X)
- Tingyu Wang (ORCID: https://orcid.org/0000-0003-2520-0511)
- Jianfeng Guo (ORCID: https://orcid.org/0009-0000-5723-6768)
Institutions
- Guangzhou University (CN)
Publication Details
- Published
- 2024-09-20
- DOI
- https://doi.org/10.1109/csis-iac63491.2024.10919437
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.00