search

amazon-science / siam-mot

SiamMOT: Siamese Multi-Object Tracking
Apache License 2.0
477 stars 61 forks source link

can this code track mutil classes object? #19

Open JAYCHOU2020 opened 3 years ago

JAYCHOU2020 commented 3 years ago

can this code track mutil classes object?

bingshuai2019 commented 3 years ago

Yes, in the demo, we provide the models that can track person as well as other multiple vehicle instances.

SherryXTChen commented 3 years ago

If we want to train the models on our own dataset with, say 3 classes, which part of the configuration should we change (and to what value)?

mondrasovic commented 2 years ago

If you want to change the number of classes to 3, then add the following entry to the configuration:

MODEL:
  ROI_BOX_HEAD:
    NUM_CLASSES: 3

I presume that you know how to properly modify the remaining configurations related to dataset path, batch size, workers and so forth.

SherryXTChen commented 2 years ago

If you want to change the number of classes to 3, then add the following entry to the configuration:

MODEL:
  ROI_BOX_HEAD:
    NUM_CLASSES: 3

I presume that you know how to properly modify the remaining configurations related to dataset path, batch size, workers and so forth.

Thank you!