ByteTrack icon indicating copy to clipboard operation
ByteTrack copied to clipboard

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

Open keqizero opened this issue 1 year ago • 5 comments

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?

keqizero avatar Mar 06 '24 15:03 keqizero

@keqizero same problem. How did you solve that?

yaoshanliang avatar May 14 '24 08:05 yaoshanliang

@keqizero same problem. How did you solve that?

No I haven't solved it

keqizero avatar May 14 '24 08:05 keqizero

@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

yaoshanliang avatar May 14 '24 08:05 yaoshanliang

@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 :)

keqizero avatar May 14 '24 09:05 keqizero

i had some reasonable success replacing the Predictor class in the palace.mp4 bytetrack demo with the Predictor class in the original yolox tools.demo.Predictor.

v-iashin avatar Jan 27 '25 11:01 v-iashin