Application of Traffic Cone Target Detection Algorithm Based on Improved YOLOv5

Abstract

To improve the automation level of highway maintenance operations, the lightweight YOLOv5-Lite-s neural network was deployed in embedded devices to assist an automatic traffic cone retractor in completing recognition and positioning operations. The system used the lightweight shuffle Net network as a backbone for feature extraction, replaced convolutional layers with focus modules to reduce computational complexity, and reduced the use of the C3 layer to increase network speed, thereby meeting the speed and accuracy requirements of traffic cone placement and retraction operations while maintaining acceptable model inference accuracy. The experimental results show that the network could maintain recognition accuracy and speed values of around 89% and 9 fps under different working conditions such as varying distances, lighting conditions, and occlusions, meeting the technical requirements for deploying and retrieving cones at a speed of 30 cones per minute when the operating vehicle’s speed was 20 km/h. The automatic traffic cone placement and retraction system operated accurately and stably, achieving the application of machine vision in traffic cone retraction operations.