Add pose_detection to output

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

View File

@ -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.