Smaller models

[lccatala]

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]

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