search

PeizeSun / TransTrack

Multiple Object Tracking with Transformer
MIT License
622 stars 109 forks source link

Is there a way to do the multi-class multi-object tracking #41

Open mehdiiranmanesh opened 2 years ago

mehdiiranmanesh commented 2 years ago

Thank you for your great work. Does this algorithm and repo support multi-class as well?

noahcao commented 2 years ago

We didn't make that as we originally target for the pedestrian-only tracking datasets, i.e. MOT datasets. But that might be trivial to support with minor modification:

  1. Following the DETR or Deformable DETR logic, we should more category for the classification.
  2. For the matching part, apart from the current matching cost, we should make sure the category is kept for the objects to be associated. But all these require you to train the model on corresponding datasets. We may make such support in the future.