search

ymzis69 / HybridSORT

[AAAI2024]Hybrid-SORT: Weak Cues Matter for Online Multi-Object Tracking
MIT License
190 stars 26 forks source link

self-trained detector #31

Open Zc229 opened 6 months ago

Zc229 commented 6 months ago

If I want to use a self-trained detector, what should I modify?

ymzis69 commented 6 months ago

For example, if you use this command: 'python tools/run_hybrid_sort_dance.py -f exps/example/mot/yolox_dancetrack_val_hybrid_sort.py -b 1 -d 1 --fp16 --fuse --expn $exp_name', then we will invoke this function for tracking: https://github.com/ymzis69/HybridSORT/blob/396f8d30db13304c0cbaf1dcf2e16ded93ce1701/yolox/evaluators/mot_evaluator_dance.py#L543, so you need to refer to the output format and replace the output of your detection model for tracking.https://github.com/ymzis69/HybridSORT/blob/396f8d30db13304c0cbaf1dcf2e16ded93ce1701/yolox/evaluators/mot_evaluator_dance.py#L661