This project provides a model for automatically drawing accurate offside-lines that can assist a referee in making an offside decision. This project aims to help smaller leagues that do not have the infastructure to implement a more sophisticated system like the one FIFA used in the 2022 World Cup in Qatar.
To install the necessary dependencies we use Poetry. After you have it installed, follow these instructions:
-
Clone this repository:
git clone [email protected]:tryolabs/soccer-video-analytics.git
-
Install the dependencies:
poetry install
First, make sure to initialize your environment using poetry shell
.
To run one of the applications (possession computation and passes counter) you need to use flags in the console.
These flags are defined in the following table:
Argument | Description | Default value |
---|---|---|
application | Set it to possession to run the possession counter or passes if you like to run the passes counter |
None, but mandatory |
path-to-the-model | Path to the soccer ball model weights (pt format) |
/models/ball.pt |
path-to-the-video | Path to the input video | /videos/soccer_possession.mp4 |
The following command shows you how to run this project.
python run.py --<application> --model <path-to-the-model> --video <path-to-the-video>
Warning: You have to run this command on the root of the project folder.
Here is an example on how to run the command:
python run.py --possession --model models/ball.pt --video videos/soccer_possession.mp4
An mp4 video will be generated after the execution. The name is the same as the input video with the suffix _out
added.