Skip to content

PelArtur/tennis-analyser

Repository files navigation

Tennis Analyser

The tennis analyser is a system of deep learning networks designed to extract statistics from input video footage. This system identifies the location of the ball, players, and court markings in the video and processes this data to generate statistics such as player movement heatmaps, ball trajectory, ball speed, and more.

This project was developed as part of the Artificial Intelligence course during the third year of the Computer Science program at the Ukrainian Catholic University.

Input Output
input1 input2

Our Team

Prerequirements

  • Our repository contains weight files for most of the models used; however, not all files could be uploaded due to GitHub restrictions. Therefore, to ensure functionality, you will need to download all the required files from this link.
  • To ensure that all libraries function correctly, you need to use Python version 3.12.X.
  • For optimal performance, we recommend running our project in a GPU-enabled environment.

How to run

  1. Clone this repository

    git clone https://github.com/PelArtur/tennis-analyser
    
  2. Install the requirements using pip

    pip install -r requirements.txt
  3. Run the following command in the command line

    python3 process_video.py -i ./videos/video1.mp4 -o ./output_video.moy

Future Plans

  • Enhance the accuracy of the bounce detector, potentially by selecting a different model.
  • Explore transferring this problem to 3D by utilizing additional tools to create a 3D reconstruction of the scene.

References

  • Yu-Chuan Huang, "TrackNet: Tennis Ball Tracking from Broadcast Video by Deep Learning Networks," Master Thesis, advised by Tsì-Uí İk and Guan-Hua Huang, National Chiao Tung University, Taiwan, April 2018.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •