Skip to content

YOLOv8 on Basketball Sports, including player detection, pose estimation.

Notifications You must be signed in to change notification settings

LittleFish-Coder/basketball-sports-ai

Repository files navigation

Basketball Sports AI

NCKU Miin Wu School of Computing Project YOLOv8 on Basketball Sports, including player detection, pose estimation.

YOLOv8 from Ultralytics

Dataset

The dataset is provided by National Cheng Kung University Women’s Basketball Team.(Private for now)

And we use Roboflow platform to label the dataset.

Player Action Detection

By training the model with custom dataset, we build a model to detect player actions.

The current model can detect actions below:

  • Shot

shot_detection

Apply on real basketball game play

gameplay_shot_detection

Pose Estimation

By using pretrained model, we can get the pose estimation result as the video below. pose_estimation_example

Future Work

  • Build a player detection model
  • Build a ball tracking model
  • Auto clip the highlight of the game from shot detection
  • Add more action detection

About

YOLOv8 on Basketball Sports, including player detection, pose estimation.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published