mikel-brostrom / boxmot

BoxMOT: pluggable SOTA tracking modules for segmentation, object detection and pose estimation models
GNU Affero General Public License v3.0
6.57k stars 1.7k forks source link

How can I use this for single class? #119

Closed ghost closed 3 years ago

ghost commented 3 years ago

I have trained yolov5 for one fruit class oranges I want to count and track the fruit on tree how can I use this repo?

mikel-brostrom commented 3 years ago

You will need to implement a counter, this repo does not contain such functionality

ccl-private commented 3 years ago

I trained a yolov5 for a custom class, can I track my custom class use this repo?

ccl-private commented 3 years ago

or should I training the re-id model my self? https://github.com/ZQPei/deep_sort_pytorch#training-the-re-id-model

ghost commented 3 years ago

author has no reply :(

mikel-brostrom commented 3 years ago

or should I training the re-id model my self? https://github.com/ZQPei/deep_sort_pytorch#training-the-re-id-model

Ideally yes. But you can start with the default tracking functionality to see whether it fulfills your tracking requirements or not.

I trained a yolov5 for a custom class, can I track my custom class use this repo?

Yes, check: https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch#filter-tracked-classes

ghost commented 3 years ago

I used the trained model of yolov5, i want to track and count the number of objects in the video but it is not working, How can it gives the random values to the objects

mikel-brostrom commented 3 years ago

Again, there is not counting functionality in this repo. You need to implement it yourself

ccl-private commented 3 years ago

@hammadyounas2008 maybe you can draw the trajectory of the object

ghost commented 3 years ago

@ccl-private How to avoid that?

ghost commented 3 years ago

@mikel-brostrom How can I implement in this track.py file?

ccl-private commented 3 years ago

@hammadyounas2008 maybe you need to implement a algorithm yourself. I also plan to count objects amount in the near future.