ZQPei / deep_sort_pytorch

MOT using deepsort and yolov3 with pytorch
MIT License
2.83k stars 727 forks source link

merge two checkpoints #18

Closed sandeshshrestha45 closed 5 years ago

sandeshshrestha45 commented 5 years ago

Dear ZQPei,

Is it possible to implement tracking two different classes in the same video? I have separately trained the re-identification model for humans and vehicles so I got two separate checkpoints. How can I track both humans and vehicles in the same video?

ZQPei commented 5 years ago

Yes, it is possible. You can implement that by creating two Deep_sort trackers to track seperately for human and vehicle class.