TransTrack icon indicating copy to clipboard operation
TransTrack copied to clipboard

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

Open mehdiiranmanesh opened this issue 4 years ago • 1 comments

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

mehdiiranmanesh avatar Sep 15 '21 00:09 mehdiiranmanesh

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.

noahcao avatar Oct 11 '21 07:10 noahcao