AI Golf: Golf Swing Discrepancy Detector

For downloading the pre-trained weights, click the link below:

Video Transformer Approach: vit_classifier_epoch_7.pth
GolfDB's SwingNet + MMPose 3D Skeleton Points Inference + ResNet: skeleton-points-white-background_classifier_epoch_6.pth

Dependencies

For more technical details, please check Project Report.

Name		Name	Last commit message	Last commit date
Latest commit History 14 Commits
golfdb-master		golfdb-master
mmpose-main		mmpose-main
sample_dataset		sample_dataset
CS_231N_Poster.png		CS_231N_Poster.png
Final_Project_Report.pdf		Final_Project_Report.pdf
README.md		README.md
pro_am_classifier.pth		pro_am_classifier.pth
swing_position_frames_model.ipynb		swing_position_frames_model.ipynb
video_transformer_model.ipynb		video_transformer_model.ipynb
youtube_downloader.ipynb		youtube_downloader.ipynb