LiuShuaiyr / UAVMOT

multi-object tracking meets moving UAV
36 stars 4 forks source link

UAVMOT: Multi-Object Tracking Meets Moving UAV

This work is built on the MCMOT, which adapts to multi-object tracking on moving UAV videos. You can refer to origin fork MCMOT.

1 train

i) Prepare dataset

Run gen_dataset_visdrone.py or gen_dataset_UAVDT.py

ii) Run train.py

2 test

Run python track.py

The multi-object tracking results saved in **.txt, and evaluate it by the official toolkits.

For visdrone2019 dataset, you can refer to the link(https://github.com/VisDrone/VisDrone2018-MOT-toolkit)

For UAVDT dataset, you can refer to the link(https://github.com/VisDrone/VisDrone2018-MOT-toolkit

3 demo

Run python demo.py