Tracking and finding coordinates for players in a youtube video
Jupyter Notebook Python
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
hypes
.gitignore
13000-.3-oobfilter.gif
Pipeline.ipynb
README.md
court.py
court_warp.py
courttracker.ipynb
data_labeling.ipynb
player_tracker_hog.ipynb
predict.py
train.py
train.pyc
youtube.py

README.md

Basketball Player Tracker

This code requires using tensorbox to create a model for training.

If you're just interested in running the notebook, a pre-trained model and the test video are located here. Place the video in the data folder and the ckpt file in your output folder

After creating the model run the steps in the pipeline with your model included on a clip of a game. Currently the model only works in clips that stay on the same half of the court.

img

For more information about the project and future work to be done check out the blog post