Skip to content

Computer Vision project to analyze and obtain data from a tennis match video.

Notifications You must be signed in to change notification settings

kaelkkd/tennis-analysis-ml

Repository files navigation

Tennis Analysis

This repository contains a Computer Vision project that analyzes and obtain data from a tennis match video.

Table of Contents

Introduction

This project is designed to provide insights into tennis matches by analyzing video footage. The project utilizes YOLO to track both players and the ball and CNNs to find the court keypoints.

Features

  • Player detection and tracking
  • Ball detection and tracking
  • Calculation of movement speed for players and the ball
  • Visualization of tracking data on the video

Installation

To run this project locally, follow these steps:

  1. Clone the repository:
    git clone https://github.com/kaelkkd/tennis-analysis.git
    cd tennis-analysis
    
  2. Install the requirements:
    gpip install requirements.txt
    
  3. Change the input at ìnput_videos and the players heights at constants as desired

Example

Screenshot from 2024-06-25 23-34-48

About

Computer Vision project to analyze and obtain data from a tennis match video.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published