SoccerNet / sn-gamestate

[CVPRW'24] SoccerNet Game State Reconstruction: End-to-End Athlete Tracking and Identification on a Minimap (CVPR24 - CVSports workshop)
240 stars 51 forks source link

Smaller models #12

Open lccatala opened 5 months ago

lccatala commented 5 months ago

Hi, would it be possible to release a smaller version of the ReID model, or at least the training code so I can train it myself?.

I've been trying to reduce the inference time but to no avail.

victorjoos commented 5 months ago

Hi @lccatala,

We trained the model with this repository and the prtreid model. The train method is here : https://github.com/SoccerNet/sn-gamestate/blob/main/sn_gamestate/reid/prtreid_api.py And more importantly the configuration for training can be found here : https://github.com/SoccerNet/sn-gamestate/blob/main/sn_gamestate/configs/modules/reid/prtreid.yaml

You should set "training_enabled" to true, and change the backbone to a smaller model, you can find more info about the training inside the prtreid repository : https://github.com/VlSomers/Prtreid