Open lccatala opened 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
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.