nwojke / deep_sort

Simple Online Realtime Tracking with a Deep Association Metric
GNU General Public License v3.0
5.3k stars 1.48k forks source link

Custom IDs for tracks #324

Closed bzand closed 7 months ago

bzand commented 7 months ago

Hi everybody

Is it possible to use custom IDs for the tracks in DeepSORT?

I am tracking a predefined number of animals in a barn. Each of these animal has been attributed a unique ID by a human beforehand. I've trained the ReID model provided for DeepSORT on these data.

My problem: By default, DeepSORT starts at track_id = 1 and increases this ID by one for each new track created. Is it possible to have custom IDs for the tracks so that they match with the ID attributed to the animals?

How would you do this?

Thanks

bzand commented 7 months ago

Hi

I wasn't properling understanding the aim of the deep appearance descriptor when I opened the issue. In the meantime it got clearer.

If I want to have other IDs than the ones provided by DeepSORT, I need to generate them, f.ex. with a classifier that I have trained on my data previously.

Cheers.