ifzhang / ByteTrack

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

A problem in the update() of byte_tracker.py #187

Closed 376498485 closed 2 years ago

376498485 commented 2 years ago

The 166th line in byte_tracker.py

if output_results.shape[1] == 5:

Normally, the detection results contains following items: x1, y1, x2, y2, score, class or x1, y1, x2, y2, score1, score2, class And I use the yolox in this rep, the "output_results.shape[1]" is 7, and the detection results of yolor is 6 So I think you can consider the "6" case. The code can be written

if output_results.shape[1] == 5 or output_results.shape[1] == 6: