search

regob / vehicle_mtmc

Vehicle MTMC Tracking
MIT License
24 stars 2 forks source link

Can I change the detector to yolov7 or other? #6

Closed spoon0525 closed 1 month ago

spoon0525 commented 1 month ago

please tell me which file I need to change, so I can use yolov7 as the devector

regob commented 1 month ago

Hi!

The detector is loaded in load_detector.py. This uses the yolov5 repo, so I guess you could add loading yolov7 here if the DETECTOR config parameter is e.g. 'yolov7'

Detection is happening here, so if your detector has a different API, this has to be changed.

spoon0525 commented 1 month ago

understand, thank you very much!