Multi-task Deep-Learning Vehicle Detection and Tracking based on Aerial Views from UAV
Cong Phuc Nguyen, Van Tuan Nguyen, Duc Dung Tran, Anh Minh Nguyen, Ngoc Phong Dao, Dinh Tuan Tran, Joohoo Lee, Anh Quang Nguyen · 2022
For vehicle management systems, vehicle detection, tracking, and recognition which provides statistical information on the number of vehicles and their characters are essential task, however, not only the vehicles themselves but also its characteristic such as types, and colors,.. is important for management. Besides, not only fixed traffic cameras, autonomous UAVs with cameras also can bring more flexibility and extend the management areas from an aerial view. In this paper, we propose and implement a real-time multi-task deep-learning for four-wheel vehicle detection, classification, and tracking system on UAVs. A dataset for detection and multi-task classification including multiple colors and types is built. To archive real-time the operation, we used the ByteTrack algorithm with YOLOv5, which is more efficient for mobile and embedded vision applications. Experimental results achieved high accuracy, more than 90% in multi-task classifying, and 56.21 HOTA tracking evaluation metrics on our created test set. The processing time is fast enough for real-time operation, 14 FPS for the detector, and 64 FPS for classification based on an embedded computer, Jetson Nano.