From b3d7599ca302bd44c80f93bcb2c3a0c0503e6675 Mon Sep 17 00:00:00 2001 From: Dave Johansen Date: Tue, 15 Nov 2022 17:03:17 -0700 Subject: [PATCH] Add pose_detection to output Solves #2582 --- mediapipe/python/solutions/pose.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mediapipe/python/solutions/pose.py b/mediapipe/python/solutions/pose.py index 74a52d611..29c2d5418 100644 --- a/mediapipe/python/solutions/pose.py +++ b/mediapipe/python/solutions/pose.py @@ -159,7 +159,7 @@ class Pose(SolutionBase): 'poselandmarkbyroicpu__tensorstoposelandmarksandsegmentation__ThresholdingCalculator.threshold': 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: """Processes an RGB image and returns the pose landmarks on the most prominent person detected.