matt-newell / disc-tracker

Disc Tracker allows you to make your own "Follow Flights". Use a tripod to record a video of your disc golf throw, then use this software to overlay the flight path of the disc

Disc Tracker

Make your own Follow Flights! This project uses OpenCV to trace the location of a disc through its flight path.

Installation

1. Clone this repo
2. Create virtual environment and install dependencies

  $ python3 -m venv venv
  $ source venv/bin/activate
  $ pip install -r requirements.txt

Usage:

1. Run program with desired video as command line argument

$ python main.py video_name.mp4

2. Press spacebar to move through video frame by frame, until the disc just leaves the throwers hand
3. Press the 'a' key, then use your mouse to draw a rectangle around the disc. Try to center the box around the center of the disc. If you are not satisifes with the location of the box, just click and drag to create a new one.
4. Press enter to confirm your bounding box, then watch as your throw is traced onto the video

Limitations

Disc Tracker only works on static videos taken with a tripod (no panning or zooming). This software may be lose track of the disc if the background is bright (i.e. the disc's path crosses the sun) or is similar in color to the disc (i.e. blue disc against blue sky). The disc may also be lost if its flight carries it too far from the camera.

Credit

The project was developed by Zachary Hislop, Lance Litten, and Nick Schnorr for CPE 428 (Intro to Computer Vision) at Cal Poly, San Luis Obispo