-
Notifications
You must be signed in to change notification settings - Fork 1
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
Pose estimate #15
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
- 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)
Mikonooooo
requested changes
Oct 21, 2023
There was a problem hiding this 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
- 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
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
see commit message for update information
implement pose detection for getting player limb angles in each frame