cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add pose_detection to output

Browse Source 
Solves #2582
This commit is contained in:
Dave Johansen 2022-11-15 17:03:17 -07:00 committed by GitHub
parent ab2dd779e7
commit b3d7599ca3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mediapipe/python/solutions/pose.py
View File

@ -159,7 +159,7 @@ class Pose(SolutionBase):
'poselandmarkbyroicpu__tensorstoposelandmarksandsegmentation__ThresholdingCalculator.threshold': 'poselandmarkbyroicpu__tensorstoposelandmarksandsegmentation__ThresholdingCalculator.threshold':
min_tracking_confidence, min_tracking_confidence,
}, },
outputs=['pose_landmarks', 'pose_world_landmarks', 'segmentation_mask']) outputs=['pose_landmarks', 'pose_world_landmarks', 'segmentation_mask', 'pose_detection'])
def process(self, image: np.ndarray) -> NamedTuple: def process(self, image: np.ndarray) -> NamedTuple:
"""Processes an RGB image and returns the pose landmarks on the most prominent person detected. """Processes an RGB image and returns the pose landmarks on the most prominent person detected.