Multi-Color Vehicle Tracking Based on Lightweight Neural Network
Mingdi Hu, Ying Li, Long Bai · 2022 4th International Conference on Natural Language Processing (ICNLP) · 2022
Vehicle tracking plays an important role in intelligent traffic management and criminal investigation assistance. At this stage, vehicle target detection technology has reached a relatively mature level of accuracy, but it is difficult to deploy to embedded platforms as the network depth increases. The computing power of the computer also has high requirements. In addition, the continuous increase of vehicle color richness increases the difficulty of target tracking, and it is impossible to identify color types that do not appear in the data set. In response to the above-mentioned problems, this paper proposes an improved YOLOv3 target detection algorithm that can be transplanted to the embedded side. Aiming at the shortcomings of the original YOLOv3 algorithm model that occupies a large amount of memory and is difficult to detect in real time on the embedded side, the lightweight MobileNetv2 depth reduce the network to replace the original YOLOv3 backbone network Darknet-53 for feature extraction, and at the same time make matching changes for anchors to adapt to the characteristics of the dataset for detection. Use the extracted self-built 24-color dataset for training, and then the experimental results show that the parameters and detection speed of the light-weighted YOLOv3 network is significantly better than that YOLOv3, and the recognition accuracy of the our dataset can reach 94.5%.