Real-Time Tennis Analytics with YOLO and CNN
This project aims to develop an AI system with YOLO for player and ball detection and a custom CNN for court mapping, delivering real-time analytics to optimise strategies and identify performance gaps
Here's a sample frame taken from one of the processed videos, demonstrating how the analysis functions:
- Player Detection Model:
- Court Key Point Model:
-
Create a Conda Environment: Run the following command in your terminal or Anaconda prompt to create a new Conda environment named
tennis_analysis
with Python 3.8:conda create -n tennis_analysis python=3.8
-
Activate the Environment:
Once created, activate your new environment using:
conda activate tennis_analysis
-
Install Dependencies:
install the required packages using the requirements.txt file:
pip install -r requirements.txt
Place your video in the input video folder and run the following command:
python main.py