Skip to content

VlSomers/prtreid

Repository files navigation

PRTreID

Multi-task Learning for Joint Re-identification, Team Affiliation, and Role Classification for Sports Visual Tracking

[Paper]

arXiv Hippocratic License HL3-LAW-MEDIA-MIL-SOC-SV

Multi-task Learning for Joint Re-identification, Team Affiliation, and Role Classification for Sports Visual Tracking, MMSports'23

Amir M. Mansourian*, Vladimir Somers*, Christophe De Vleeschouwer, Shohreh Kasaei

arxiv 2401.09942

PRTreID diagram

PRTreID is based on "BPBReID: Body Part-based (Occluded) Re-Identification": a lot of relevant information can be found in the original repository. This PRTreID repository will be updated soon with detailed instructions on how to run the model and download the weights. Compared to the original BPBreID repository, a new head was added to the main model to predict roles, and two new training objectives were added to train the model on team affiliation and role classification. PRTreID was tested on the SoccerNet-GSR and SoccerNet-Tracking datasets. To train/run PRTreID on SoccerNet, please have a look at this repository: a dataloader is provided to turn the SoccerNet dataset into a Torchreid compatible dataset, to be used by this repository for training PRTreID. If you need to run PRTreID, please install SoccerNet-GSR, carefully read the instructions on how to use the codebase, and enable PRTreID training by setting the "training_enabled" to True inside prtreid.yaml. Once SoccerNet-GSR is run with the above config enabled, a reid dataset will be automatically extracted from the soccernet tracking dataset, and PRTreID will be subsequently trained on the generated dataset. If you need to change any training config, please do so inside prtreid.yaml: this .yaml file contains Torchreid compatible configurations, as described here and here. In practice, prtreid.yaml is a mirror of the config defined in default_config.py. The current description and code will be improved soon to ease installation, please follow the repository to stay tuned. If you have any questions or need further guidance to install PRTreID, feel free to open a GitHub issue.

News

  • [2024.01.1] Code release

Citation

If you use this repository for your research or wish to refer to our method PRTreID, please use the following BibTeX entry:

@inproceedings{Mansourian2023Multitask,
	title = {Multi-task Learning for Joint Re-identification, Team Affiliation, and Role Classification for Sports Visual Tracking},
	author = {Mansourian, Amir M. and Somers, Vladimir and De Vleeschouwer, Christophe and Kasaei, Shohreh},
	booktitle = mmsports,
	shortjournalproceedings = {Proc. 6th Int. Work. Multimedia Content Anal. Sports},
	pages = {103–112},
	month = Oct,
	year = {2023},
	publisher = {ACM},
	address = city-ottawa,
	keywords = {SoccerNet, Bottom-up, ARTHuS, Team affiliation, Role classification},
	doi = {10.1145/3606038.3616172},
	url = {https://doi.org/10.1145/3606038.3616172}
}

Acknowledgement

This codebase is a fork from BPBreID

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages