Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Pose estimate #15

Merged
merged 30 commits into from
Oct 25, 2023
Merged

Pose estimate #15

merged 30 commits into from
Oct 25, 2023

Conversation

dweizzz
Copy link
Collaborator

@dweizzz dweizzz commented Oct 21, 2023

see commit message for update information
implement pose detection for getting player limb angles in each frame

- Create branch
- Write src/pose_estimation/pose_estimate.py
- Works, but does not do anything right now
- When running, terminal outputs non fatal errors (?), need to test
- Update requirements for Pose Estimation
- Add comments
- Clean code (get rid of prints, format output json)
- Changed to predict on whole video, process entire video
- Created Pose function in modelrunner and call in main (runs results once)
- Add trained model (best.pt)
@dweizzz dweizzz requested a review from ericguo31 October 21, 2023 15:51
Copy link
Collaborator

@Mikonooooo Mikonooooo left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Fix keypoint output in pose estimate
Move pose and weights into own folder

dweizzz and others added 23 commits October 21, 2023 13:16
- checks if left wrist above left shoulder and right wrist above right shoulder
- also counts some non-shots (ex: rebounds)
- writes frame of attempted shot and person in tmp/shots.txt
- to ensure ffmpeg is installed for macos when reencoding video
- Create two processes to track person and ball separately
- Commented lines of code for multithreading
- ShotAttempt tracks start/end frame, player_id, and ShotType
- did not change any functional code
- second parse ball function in general.detect
- states now contains "balls" : {ball[id] : BallState}
- states represented as dictionary of Frames
- Frame represented as object of players and balls dictionaries
- code not yet tested
- separate parsing outputs into GameState into parse.py
- update frame-by-frame info on players, balls, and rim
- not tested yet
- track universal stats in players and balls dictionaries
- not tested yet
- apply spectral clustering to solve sparest cut
- separate possession and pass calculation out for orthogonality
- with sliding window for detection
- adjusted ci to run processing
@Mikonooooo Mikonooooo merged commit 8e4e174 into ball-202 Oct 25, 2023
@Mikonooooo Mikonooooo deleted the pose-estimate branch October 25, 2023 04:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants