This is the repo for multi-object tracking based on YOLOv3. we focus on achieving a real-time tracking system which integrates state-of-art object detection algorithm and multi object tracking algorithm into single one system. This can take us one step further to realizing a fully automatic and highly intelligent security system. We also apply the system on UAV to achieve the real-time multi object tracking on UAV video streaming.
Hao(Alex) Xiao, Haotian Zhang, Sujie Zhu
The system can be roughly divided into two mainly part, which are object detection part and multi object tracking part. First, we will use the WIFI to stream the video from the drone to the computer. Then we apply object detection frame by frame. For every object in one frame, we extract the appearance feature and motion feature and send these information into multi object tracking part.
- OpenCV
- CUDA
- cuDNN
- Install
git clone https://github.com/AlexXiao95/YOLO_TRACKING
cd YOLO_TRACKING
make
- Download the pre-trained weight file:
wget https://pjreddie.com/media/files/yolov3.weights
- Run real-time tracking on webcam:
./darknet detector demo cfg/coco.data cfg/yolov3.cfg yolov3.weights -c <cam_id>
- Run real-time tracking on video file:
./darknet detector demo cfg/coco.data cfg/yolov3.cfg yolov3.weights <video_file>
- Run real-time tracking on drone video (Should set up video stream first):
./darknet detector demo cfg/coco.data cfg/yolov3.cfg yolov3.weights bebop.sdp
For more information about darknet and YOLO, please refer to Darknet project website.
For questions or issues please use the Google Group.