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

Media Pipe Pose Estimation + Visualization #203

Merged
merged 26 commits into from
Jan 22, 2025
Merged
Changes from 1 commit
Commits
Show all changes
26 commits
Select commit Hold shift + click to select a range
90d014f
Support for media pipe operations
brukew Nov 8, 2024
206eed0
Test Suite for mediapipe pose estimation + docstrings
brukew Nov 19, 2024
83c31b7
reset changes from main
brukew Nov 19, 2024
269f0a6
Refactored into data structures and tasks + generalized visualization
brukew Nov 24, 2024
b83207a
Merge remote-tracking branch 'origin/main' into 173-task-pose-estimation
brukew Jan 2, 2025
bb0efb1
Revision of pose estimation DS + addition of YOLO
brukew Jan 7, 2025
9602835
Removed model from package, downloads as needed
brukew Jan 8, 2025
641aef5
Docs for pose estimation
brukew Jan 12, 2025
b28b77c
Tutorial for pose estimation
brukew Jan 12, 2025
f639ff2
Implement visualization for mediapipe and yolo
brukew Jan 12, 2025
400ae23
Visualization added to tests and tutorial
brukew Jan 13, 2025
f54850c
Created api for pose estimation
brukew Jan 13, 2025
fc20beb
Update pose estimation init
brukew Jan 13, 2025
432c4d1
Corrected tutorial
brukew Jan 13, 2025
00b3676
Edited tests to use the api
brukew Jan 15, 2025
4057794
FIxed test file paths
brukew Jan 15, 2025
3600eca
Uodated tutorial with api and more info
brukew Jan 15, 2025
32bc0a1
Merge branch 'main' into 173-task-pose-estimation
fabiocat93 Jan 16, 2025
9e2c82e
Update pyproject.toml - dependency versioning with ~=
fabiocat93 Jan 16, 2025
294739c
Plot image with visualize
brukew Jan 18, 2025
68f1857
Fixed visualization
brukew Jan 21, 2025
8d646b0
Unified landmarks + visualization
brukew Jan 21, 2025
216c80b
Remove test block in api
brukew Jan 21, 2025
1193edb
Remove plot visualization - sanity check
brukew Jan 21, 2025
352b082
Moved visualization plot out of api
brukew Jan 21, 2025
8458609
Optionally plot pose image instead of auto.
brukew Jan 21, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Remove plot visualization - sanity check
  • Loading branch information
brukew committed Jan 21, 2025
commit 1193edbdde86117277ad1c3f456b06d38013516c
7 changes: 3 additions & 4 deletions src/senselab/video/tasks/pose_estimation/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@

from typing import Any, Optional

import matplotlib.pyplot as plt
import numpy as np

from senselab.video.data_structures.pose import ImagePose
Expand Down Expand Up @@ -73,8 +72,8 @@ def visualize_pose(pose_image: ImagePose, output_path: Optional[str] = None) ->
"""
annotated_image = visualize(pose_image, output_path=output_path)

plt.imshow(annotated_image)
plt.axis("off")
plt.show()
# plt.imshow(annotated_image)
# plt.axis("off")
# plt.show()

return annotated_image
Loading