Is it possible to run deepsort tracker with osnet?

ml6team / deepstream-python

NVIDIA Deepstream 6.1 Python boilerplate

MIT License

130 stars 36 forks source link

Is it possible to run deepsort tracker with osnet? #12

Closed YoungjaeDev closed 1 year ago

YoungjaeDev commented 1 year ago

Hi! Can I replace the cosine_metric model provided by deepsort based on the osnet provided?

julestalloen commented 1 year ago

The re-identification model used by DeepSORT is specified here: https://github.com/ml6team/deepstream-python/blob/master/deepstream/configs/trackers/DeepSORT_config.yml#L74

You could try changing that (and related settings). I have not tried it myself yet. The re-identification pipeline in this repository uses OSNet as a secondary model, in addition to the DeepSORT mars-small128.