Ball + Player Tracking

This project runs ball tracking (TrackNet) and player tracking (YOLOv8 + Kalman filter) together on a video.
It produces two outputs: - output_marked.mp4 → video with players and ball marked - output_mask.mp4 → mask video (players & ball in white on black)-

How to Run

1. Clone this repo and move into the folder:

git clone https://github.com/Komantxe/PadelCheeseAI.git
cd PadelCheeseAI

2. Create a virtual environment

python -m venv venv
source venv/bin/activate   # On Windows: venv\Scripts\activate

3. Install dependencies:

pip install -r requirements.txt

4. Edit the script to set your input/output video paths:
   Open track.py and update:

VIDEO_PATH = "input.mp4"   # input video here
VIDEO_OUT  = "output_marked.mp4"      # annotated output
MASK_OUT   = "output_mask.mp4"        # mask video output

6. Run the script:

python track.py

---

✅ Output

• Annotated video → output_marked.mp4
• Mask video → output_mask.mp4