ifzhang / ByteTrack

[ECCV 2022] ByteTrack: Multi-Object Tracking by Associating Every Detection Box
MIT License
4.8k stars 905 forks source link

Using my own YOLOX model trained with official YOLOX repo, but demo_track.py outputs nothing #392

Open keqizero opened 8 months ago

keqizero commented 8 months ago

Hi, thank you for your great work!

I have a question about the YOLOX model. I am trying to achieve better results by replacing the YOLOX model with my own, which was trained using official YOLOX repo. However, after changing the model, I don't see any detection nor tracking results at all. The saved txt contains nothing. I assume it's because of the different model preparation, but I couldn't fix it.

Do you have any idea what causes the problem?

yaoshanliang commented 6 months ago

@keqizero same problem. How did you solve that?

keqizero commented 6 months ago

@keqizero same problem. How did you solve that?

No I haven't solved it

yaoshanliang commented 6 months ago

@keqizero I find official YOLOX ends with .pth, while ByteTrack ends with .pth.tar. Evaluation detection results from official YOLOX and ByteTrack are different. Thus, I am confusing on using which detector

keqizero commented 6 months ago

@keqizero I find official YOLOX ends with .pth, while ByteTrack ends with .pth.tar. Evaluation detection results from official YOLOX and ByteTrack are different. Thus, I am confusing on using which detector

I think the saving format shouldn't be a problem. It's just a name. The problem is the difference of the data preparation or model usage. I haven't looked into the code for a long time. I hope you have better luck :)