YOLOv5s-Based Lightweight Object Recognition with Deep and Shallow Feature Fusion
In object detection, targets in adverse and complex scenes often have limited information and pose challenges for feature extraction. To address this, we designed a lightweight feature extraction network based on the Convolutional Block Attention Module (CBAM) and multi-scale information fusion. Within the YOLOv5s backbone, we construct deep feature maps, integrate CBAM, and fuse high-resolution shallow features with deep features. We also add new output heads with distinct feature extraction structures for classification and localization, significantly enhancing detection performance, especially under strong light, nighttime, and rainy conditions. Experimental results show superior detection performance in complex scenes, particularly for pedestrian crossing detection in adverse weather and low-light conditions. Using an open-source dataset from Shanghai Jiao Tong University, our algorithm improves pedestrian crossing-detection precision (AP0.5:0.95) by 5.9%, reaching 82.3%, while maintaining a detection speed of 44.8 FPS, meeting real-time detection requirements. The source code is available at GitHub.
Authors
- Lin Xu (ORCID: https://orcid.org/0000-0001-6952-824X)
- Guili Wang (ORCID: https://orcid.org/0000-0001-8482-256X)
- Longlong Tian (ORCID: https://orcid.org/0000-0002-9343-6561)
- Chang Liu (ORCID: https://orcid.org/0000-0002-5187-5507)
- Liguo Qu (ORCID: https://orcid.org/0000-0001-8848-9784)
- Hangyu Zhang (ORCID: https://orcid.org/0000-0002-5392-8019)
- Minyu Zhou (ORCID: https://orcid.org/0000-0003-2596-2778)
- Liangwang Sun
- Chenhao Li (ORCID: https://orcid.org/0009-0005-8903-0955)
Institutions
- Anhui Normal University (CN)
Publication Details
- Journal
- Electronics
- Published
- 2025-02-28
- DOI
- https://doi.org/10.3390/electronics14050971
- Citations
- 1
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 1.28
Funders
- National Natural Science Foundation of China