Ball tracking in volleyball
This repo is for ball recognition and tracking in live volleyball game.
Requirements
- Python3
- OpenCV
- Keras with Tensorflow
How to use
- Get a video file with a game fragment
- Get highest blobs:
python3 high.py <path_to_vb_file> <mask_output_dir> <color_output_dir>
- Classify manually the blobs into 2 classes (b)all/(n)ot ball
- Put the classified data into vball-net/train
- cd vball-net
- Python3 train.py
- Test blobber:
*python3 blobber.py <path_to_vb_file>
You will see and output with ball paths like that:
- Run a player with the live ball tracking:
*python3 ball_play.py <path_to_vb_file>
Links
- Used an open dataset from some austrian league
- Story @ TowardDataScience
- Story @ Habr(Russian)
- vball.io - a service I started to cut rallies and digest volleball videos based on this ball tracking algo.