Vehicle identification and analysis based on lightweight yolov5 on edge computing platform

Abstract With the accelerating process of urbanization, the role of intelligent transportation systems in urban traffic management has become more crucial. Nevertheless, traditional surveillance cameras only possess the ability to capture videos and lack the capacity for vehicle detection. Consequently, in order to enhance the intelligent capabilities of front-end surveillance cameras and achieve lightweight model deployment, this research has designed a lightweight detection model that can be deployed on embedded devices. Deploying detection models on embedded devices requires a high level of storage and system computing power, and there exist issues such as model speed, model accuracy, model size, and energy efficiency that are difficult to address in existing studies.Specifically, based on YOLOv5, we first use lightweight structures to replace the C3 module and introduce GSConv and Slim Neck structures to replace the Neck of YOLOv5 in order to reduce the model's parameters and computational complexity. After replacing the model structure, we use global channel pruning to remove redundant information, and use asymmetric quantization to convert model parameters from floating point numbers to fixed point numbers, significantly reducing the model's size and load time, and improving the inference speed. For the target tracking part, we combine the ByteTrack target tracking algorithm with the proposed model, effectively reducing the impact of target occlusion on detection and improving the correlation between video frames. In terms of data recording, detection information such as vehicle color, license plate number, and road blockage level can be uploaded to the database server, and the model's detection results can be encoded using the H.264 video encoding format and uploaded to the streaming media server through streaming media protocols. In the experimental part, we selected the embedded Rk3399pro as the deployment object and achieved 97.8% accuracy in the vehicle counting task, meeting the real-time performance requirements of 31FPS.