This repository uses YOLO and Kalman Filters to track a single and multiple moving balls and predict the center of the ball
The code is located at main.ipynb with input being single_ball.mp4 and multiple_balls.avi. After running the notebook, output has been stored in single_ball_kalman.mp4 and multiple_balls_kalman.mp4