Use open cv to detect critical elements in a golf swing.
Project description
Table of Contents
- Introduction Purpose of the package.
- Installation If you wish to use the package.
- Development If you wish to change the source code and enhance the package.
Introduction
The package provides various scripts to process a golf swing video and export the information to another package "GolfCoach" for analysis. The output feedback can then be used to improve the swing.
The package uses Google media pipe for detecting the golfer. It then detects the club head in the video and the golf poses to detect the start and finish of the golf swing. This information in written out to a json file that can be analyzed by a separate packge "GolfCoach" for analysis.
This is shown below
Some sample vides for down the line, right hand golfer is available on GoogleDrive
Each video example has 3 files.
-
Video file
-
Pickle database that is the result of running the scripts below.
-
Json file for exporting to GolfCoach packge
The general sequence of steps for a video say "0001.mov"
Create Swing Database
This step runs google media pipe and stores all the landmarks. It will also try to detect the golf poses and club head position. Current version usually fails on this and requires hand labellling.
create_swing_db 00001.mov
Hand Label GolfPoses
Run the script below to label the golf poses. The golf pose consist of
- Start Pose:
- Top Pose: Indicates that the backswing is now complete.
- Finish Pose: Indicates that the swing is complete.
label_golf_poses 00001.pkl
Hand Label Club Head Position
Label the club head in some of the frames. The script will then use a linear approx to fill the position in the remaining frames.
label_club_head 00001.pkl
Export To Json
Export the information to a json file for analysis by the GolfCoach package.
dump_swing_db 00001.pkl
Status
WIP: Detailed src documentation of this package is on ReadTheDocs
Status
The current steps are listed here GolfSwingCapture Initial thoughts on swing analysis are here GolfSwingAnalysis
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distribution
Hashes for golftracker-2.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7067db7fb6862eb09d62e9487fd862011ec740aa84c69ae3e2b2ed9fb15ad20a |
|
MD5 | 057c5b1df8a92de41ed2debf3e590b53 |
|
BLAKE2b-256 | 67a5993788772739ed3b78c058b70920f2bd2696f7d16e915062e110df9eb88a |