Apologies if this is a noob question, but I wonder how easy it would be to adapt this to other sports, eg cricket. Is the model specific to baseball, or can it be used for any ball-tracking?
I would say that the similar workflow could be applied to any ball-related sports. The object detection and the tracking algorithm is basically the same. Then, you could add any sport-specific feature!
For example, I have used a similar method to build AI Basketball Analysis.
Thanks for asking! The only model I used in this project is the YOLOv4 object detection model to detect the ball in each frame. I collected about 200 images to train it.
For the other parts like the tracking and the overlay timing, I programmed it by myself.
I implemented SORT algorithm for tracking the ball and some programming logic to capture the overlay timing from each clip.
9 comments
[ 5.1 ms ] story [ 23.9 ms ] threadI would say that the similar workflow could be applied to any ball-related sports. The object detection and the tracking algorithm is basically the same. Then, you could add any sport-specific feature!
For example, I have used a similar method to build AI Basketball Analysis.
https://github.com/chonyy/AI-basketball-analysis
https://trajektsports.com/
For the other parts like the tracking and the overlay timing, I programmed it by myself.
I implemented SORT algorithm for tracking the ball and some programming logic to capture the overlay timing from each clip.