/ball-tracking

A program that tracks the ball and estimates the shot location of a basketball shot.

Primary LanguagePython

Ball Tracking and Shot Location Estimation

A program that tracks the ball and estimates the shot location of a basketball shot.

The program works in this way:

      1.Select 4 points for source destinations for homography.
      2.Select ROI includes hoop.
      3.Watch for a motion inside the ROI.
      4.Validate the motion. With Blob Detection, distinguish the ball.
      5.Track the ball 30-35 frames back to the beginning of the shot.
      6.Find the shooter with YOLO.
      7.Calculate homography between source image and destination image.
      8.Convert world shot location points to the pixel shot location points with homography.
      9.Mark the shot location on the heat map.

Thanks to the PyImageSearch for showing me the best examples and sources.

Sample Images

alt text

alt text

alt text

alt text