SE-Lightweight YOLO: Higher Accuracy in YOLO Detection for Vehicle Inspection
Source abstract
Abstract
Against the backdrop of ongoing urbanization, issues such as traffic congestion and accidents are assuming heightened prominence, necessitating urgent and practical interventions to enhance the efficiency and safety of transportation systems. A paramount challenge lies in realizing real-time vehicle monitoring, flow management, and traffic safety control within the transportation infrastructure to mitigate congestion, optimize road utilization, and curb traffic accidents. In response to this challenge, the present study leverages advanced computer vision technology for vehicle detection and tracking, employing deep learning algorithms. The resultant recognition outcomes provide the traffic management domain with actionable insights for optimizing traffic flow management and signal light control through real-time data analysis. The study demonstrates the applicability of the SE-Lightweight YOLO algorithm, as presented herein, showcasing a noteworthy 95.7% accuracy in vehicle recognition. As a prospective trajectory, this research stands poised to serve as a pivotal reference for urban traffic management, laying the groundwork for a more efficient, secure, and streamlined transportation system in the future. To solve the existing vehicle detection problems in vehicle type recognition, recognition and detection accuracy need to be improved, alongside resolving the issues of slow detection speed, and others. In this paper, we made innovative changes based on the YOLOv7 framework: we added the SE attention transfer mechanism in the backbone module, and the model achieved better results, with a 1.2% improvement compared with the original YOLOv7. Meanwhile, we replaced the SPPCSPC module with the SPPFCSPC module, which enhanced the trait extraction of the model. After that, we applied the SE-Lightweight YOLO to the field of traffic monitoring. This can assist transportation-related personnel in traffic monitoring and aid in creating big data on transportation. Therefore, this research has a good application prospect.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Advanced Neural Network Applications · Traffic Prediction and Management Techniques · Autonomous Vehicle Technology and Safety
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
School of Information Science and Engineering, Shandong Agricultural University, Taian 271018, China
Funding
Shandong Provincial Natural Science Foundation, China · Innovation and Entrepreneurship Training Program for Students of Shandong Agricultural University
References (37)
Show source-provided references
1. Intelligent Transportation Systems10.1109/MVT.2009.935537 ↗
2. Zeng, Y. (2020). Optimal Control and Application of Traffic Light Timing Based on Fuzzy Control. [Master’s Thesis, Changsha University of Technology].
3. (2023, October 26). Issues Report: Smart Transportation Market. Manufacturing Close-Up. Available online: https://www.researchandmarkets.com/.
4. Cao, Z.W. (2023). Research on highway congestion mitigation technology based on intelligent transport. Intell. Build. Smart City, 168–170.
5. Vehicle emission control on road with temporal traffic information using deep reinforcement learning10.1016/j.ifacol.2020.12.1988 ↗
6. A unified framework for vehicle rerouting and traffic light control to reduce traffic congestion10.1109/TITS.2016.2613997 ↗
7. YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
8. Li, Z., Yuan, J., Li, G., Wang, H., Li, X., Li, D., and Wang, X. (2023). RSI-YOLO: Object Detection Method for Remote Sensing Images Based on Improved YOLO. Sensors, 23.10.3390/s23146414 ↗
9. Yang, W., Tang, X., Jiang, K., Fu, Y., and Zhang, X. (2023). An Improved YOLOv5 Algorithm for Vulnerable Road User Detection. Sensors, 23.10.3390/s23187761 ↗
10. Sang, J., Wu, Z., Guo, P., Hu, H., Xiang, H., Zhang, Q., and Cai, B. (2018). An Improved YOLOv2 for Vehicle Detection. Sensors, 18.10.3390/s18124272 ↗
11. Wang, Y., Guan, Y., Liu, H., Jin, L., Li, X., Guo, B., and Zhang, Z. (2023). VV-YOLO: A Vehicle View Object Detection Model Based on Improved YOLOv4. Sensors, 23.10.3390/s23073385 ↗
12. Song, W., and Suandi, S.A. (2023). TSR-YOLO: A Chinese Traffic Sign Recognition Algorithm for Intelligent Vehicles in Complex Scenes. Sensors, 23.10.3390/s23020749 ↗
13. Vehicle Classification and Counting System Using YOLO Object Detection Technology10.18280/ts.380419 ↗
14. Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition10.1109/TPAMI.2015.2389824 ↗
15. Wang, Z., Zhang, X., Li, J., and Luan, K. (2021). A YOLO-Based Target Detection Model for Offshore Unmanned Aerial Vehicle Data. Sustainability, 13.10.3390/su132312980 ↗
16. Qiu, Z., Bai, H., and Chen, T. (2023). Special Vehicle Detection from UAV Perspective via YOLO-GNS Based Deep Learning Network. Drones, 7.10.3390/drones7020117 ↗
17. Fast-Yolo-Rec: Incorporating Yolo-Base Detection and Recurrent-Base Prediction Networks for Fast Vehicle Detection in Consecutive Images10.1109/ACCESS.2022.3221942 ↗
18. Liao, L., Luo, L., Su, J., Xiao, Z., Zou, F., and Lin, Y. (2023). Eagle-YOLO: An Eagle-Inspired YOLO for Object Detection in Unmanned Aerial Vehicles Scenarios. Mathematics, 11.10.3390/math11092093 ↗
19. Li, Y., Wang, J., Huang, J., and Li, Y. (2022). Research on Deep Learning Automatic Vehicle Recognition Algorithm Based on RES-YOLO Model. Sensors, 22.10.3390/s22103783 ↗
20. T-YOLO: Tiny Vehicle Detection Based on YOLO and Multi-Scale Convolutional Neural Networks10.1109/ACCESS.2021.3137638 ↗
21. Vehicle Re-Identification for Automatic Video Traffic Surveillance
22. A survey of advances in vision-based vehicle re-identification10.1016/j.cviu.2019.03.001 ↗
23. A comparative study of image low level feature extraction algorithms10.1016/j.eij.2013.06.003 ↗
24. A Mathematical Model for Analyzing and Identifying the Composition of Ancient Glass Objects and Its Application
25. Niu, C., Hou, H., Shen, Y., and Zhou, Z. (2023, January 15–17). The listing price prediction of used sailboats based on LM-BP Neural Network. Proceedings of the 2023 IEEE 7th Information Technology and Mechatronics Engineering Conference (ITOEC), Chongqing, China.10.1109/ITOEC57671.2023.10292038 ↗
26. Ye, J., Yuan, Z., Qian, C., and Li, X. (2022). CAA-YOLO: Combined-Attention-Augmented YOLO for Infrared Ocean Ships Detection. Sensors, 22.10.3390/s22103782 ↗
27. A lightweight SE-YOLOv3 network for multi-scale object detection in remote sensing imagery10.1142/S0218001421500373 ↗
28. Hou, Q., Zhou, D., Feng, J., Hou, Q., Zhou, D., and Feng, J. (2021). Coordinate Attention for Efficient Mobile Network Design. Comput. Vis. Pattern Recognit., 13713–13722.10.1109/CVPR46437.2021.01350 ↗
29. Ren, S., He, K., Girshick, R., and Sun, J. (2015). Faster R-CNN: Towards real-time object detection with region proposal networks. arXiv.
30. MEB-YOLO: An Efficient Vehicle Detection Method in Complex Traffic Road Scenes
31. Tai, S.-K., Dewi, C., Chen, R.-C., Liu, Y.-T., Jiang, X., and Yu, H. (2020). Deep Learning for Traffic Sign Recognition Based on Spatial Pyramid Pooling with Scale Analysis. Appl. Sci., 10.10.3390/app10196997 ↗
32. Iraqi License Plate Recognition System Using (YOLO) with SIFT and SURF Algorithm
33. Widyastuti, R., and Yang, C.K. (2018, January 9–12). Cat’s nose recognition using you only look once (YOLO) and scale-invariant feature transform (SIFT). Proceedings of the 2018 IEEE 7th Global Conference on Consumer Electronics (GCCE), Nara, Japan.10.1109/GCCE.2018.8574870 ↗
34. Multi-small target detection and tracking based on improved YOLO and SIFT for drones
35. Wu, C., Ye, M., Zhang, J., and Ma, Y. (2023). YOLO-LWNet: A Lightweight Road Damage Object Detection Network for Mobile Terminal Devices. Sensors, 23.10.3390/s23063268 ↗
36. YOLO-LRDD: A lightweight method for road damage detection based on improved YOLOv5s10.1186/s13634-022-00931-x ↗
37. Huang, Z., Wu, J., Su, L., Xie, Y., Li, T., and Huang, X. (2023). SP-YOLO-Lite: A Lightweight Violation Detection Algorithm Based on SP Attention Mechanism. Electronics, 12.10.3390/electronics12143176 ↗