🏀 Analyze basketball shots with machine learning!
This is an artificial intelligence application built on the concept of object detection. Analyze basketball shots by digging into the data collected from object detection. We can get the result by simply uploading files to the web App, or submitting a POST request to the API. Please check the features below.
These instructions will get you a copy of the project up and running on your local machine.
Get a copy of this project by simply running the git clone command.
git clone https://github.com/chonyy/AI-basketball-analysis.git
Before running the project, we have to install all the dependencies from requirements.txt
pip install -r requirements.txt
Last, get the project hosted on your local machine with a single command.
python app.pyThis project is also hosted on Heroku. However, the heavy computation of TensorFlow may cause Timeout error and crash the app (especially for video analysis). Therefore, hosting the project on your local machine is more preferable.
This project has three main features, shot analysis, shot detection, detection API.
Counting shooting attempts and missing, scoring shots from the input video. Detection keypoints in different colors have different meanings listed below:
- Blue: Detected basketball in normal status
- Purple: Undetermined shot
- Green: Shot went in
- Red: Miss
Detection will be shown on the image. The confidence and the coordinate of the detection will be listed below.
Get the JSON response by submitting a POST request to (./detection_json) with "image" as KEY and input image as VALUE.
The object detection model is trained with the Faster R-CNN model architecture, which includes pretrained weight on COCO dataset. Taking the configuration from the model architecture and train it on my own dataset.
- Host it on azure web app service.
- Improve the efficiency, making it executable on web app services.