request for some info about FairMOT

CaptainEven / MCMOT

Real time one-stage multi-class & multi-object tracking based on anchor-free detection and ReID

MIT License

383 stars 82 forks source link

request for some info about FairMOT #15

Closed SamihaSara closed 3 years ago

SamihaSara commented 3 years ago

Hi, thanks for extending this project further. I am currently debugging the source codes. Could you please tell me which folders/files under FairMOT/src/ are performing Re-ID and Detection and while files contain heatmap, center-offset, and box-size branches?

CaptainEven commented 3 years ago

@SamihaSara Detection and ReID are performed in src/lib/tracker/multitracker.py, update_tracking and update_detection function. For training, the GT of heatmap, center-offset, box-size are prepared in src/lib/datasets/dataset/jde.py, JointDataset class.