MCG-NJU / MixSort

[ICCV2023] MixSort: The Customized Tracker in SportsMOT
MIT License
70 stars 8 forks source link

How to run inference on videos directly #12

Open usman-bsee opened 5 months ago

usman-bsee commented 5 months ago

Dear Author, Thanks for your interest, Please can you guide how to run inference on new videos without loading datasets in the dataset directory. Thanks.

ret-1 commented 5 months ago

Hello,

In inference, the get_eval_loader function defined in the exp file is used to get the DataLoader: https://github.com/MCG-NJU/MixSort/blob/a078f5bf6ae9fbeecbc1384479d5f02ab8b9e7f6/tools/track_mixsort.py#L175-L183

Since we haven't provided a script to directly run inference on custom videos, you need to manually convert it to coco format, create a new exp file, and modify the get_eval_loader function to correctly read your data.

You could refer to yolox_x_sportsmot.py and convert_sportsmot_to_coco.py