search

nwojke / deep_sort

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

Use for objects other than pedestrian #299

Closed pouryajafarzadeh closed 1 year ago

pouryajafarzadeh commented 1 year ago

Can we use the method for other objects such as cars, bikes?

Dipankar1997161 commented 1 year ago

yes you can use it for any object. The class file needs to be updated accordingly.

SalmaZakaria commented 1 year ago

@Dipankar1997161 How to update the class and how can I get it?

Dipankar1997161 commented 1 year ago

@Dipankar1997161 How to update the class and how can I get it?

The class file will most probably be there in the Architecture you using. For example if you plan to Yolo with Deep sort, Change the class file in Yolo. There is a file or a program where the classes are defined.

Or else check the latest pull request here, someome has added a detector class. That might help

SalmaZakaria commented 1 year ago

Ok, I am using deep sort with yolo an I want to return the class predicted from track objects ? is that available? For example: tracked_objects = mot_tracker.update_tracks(detections.cpu(), frame) tracked_objects contains the track id and bbox, how to make it contain the class too?

Dipankar1997161 commented 1 year ago

Ok, I am using deep sort with yolo an I want to return the class predicted from track objects ? is that available? For example: tracked_objects = mot_tracker.update_tracks(detections.cpu(), frame) tracked_objects contains the track id and bbox, how to make it contain the class too?

tools/generate_detections.py

Check the Image Encoder Class in this file. It's here only. Over there is a file for custom detections. Hope that helps