How to train the tracker?

ifzhang / ByteTrack

[ECCV 2022] ByteTrack: Multi-Object Tracking by Associating Every Detection Box

MIT License

4.69k stars 891 forks source link

How to train the tracker? #213

Open starhou opened 2 years ago

starhou commented 2 years ago

The example of Train in README is like that,

cd <ByteTrack_HOME>
python3 tools/train.py -f exps/example/mot/yolox_x_ablation.py -d 8 -b 48 --fp16 -o -c pretrained/yolox_x.pth

I debugged the code and found that only the detection part of the model was trained, how to train the whole? In other words, how do I get my own bytetrack_ablation.pth.tar

parthmalpathak commented 2 years ago

Hi, I think you get your bytetrack_ablation.pth.tar after you train. It is stored in YOLOX_Outputs directory.

zzh805780186 commented 2 years ago

Hi，Did you meet this problem when runing train.py with 'python3 tools/train.py -f exps/example/mot/yolox_x_ablation.py -d 8 -b 48 --fp16 -o -c pretrained/yolox_x.pth' ? problem