From 787371cfbac4429dbf8f09765d6940e7c38a245b Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Wed, 27 Sep 2023 12:26:06 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 568927297 --- mediapipe/python/BUILD | 1 + mediapipe/python/solution_base.py | 7 +++++++ 2 files changed, 8 insertions(+) diff --git a/mediapipe/python/BUILD b/mediapipe/python/BUILD index 085fbc96bc..b6e1ab945f 100644 --- a/mediapipe/python/BUILD +++ b/mediapipe/python/BUILD @@ -52,6 +52,7 @@ pybind_extension( "@stblib//:stb_image", # Type registration. "//mediapipe/framework:basic_types_registration", + "//mediapipe/framework/formats:body_rig_registration", "//mediapipe/framework/formats:classification_registration", "//mediapipe/framework/formats:detection_registration", "//mediapipe/framework/formats:landmark_registration", diff --git a/mediapipe/python/solution_base.py b/mediapipe/python/solution_base.py index 020d163716..ad57ef07c3 100644 --- a/mediapipe/python/solution_base.py +++ b/mediapipe/python/solution_base.py @@ -40,6 +40,7 @@ class contains the shared logic among the high-level Solution APIs including from mediapipe.calculators.util import logic_calculator_pb2 from mediapipe.calculators.util import thresholding_calculator_pb2 from mediapipe.framework import calculator_pb2 +from mediapipe.framework.formats import body_rig_pb2 from mediapipe.framework.formats import classification_pb2 from mediapipe.framework.formats import detection_pb2 from mediapipe.framework.formats import landmark_pb2 @@ -192,6 +193,12 @@ def from_registered_name(registered_name: str) -> 'PacketDataType': PacketDataType.PROTO_LIST, '::std::vector<::mediapipe::NormalizedRect>': PacketDataType.PROTO_LIST, + '::mediapipe::Joint': + PacketDataType.PROTO, + '::mediapipe::JointList': + PacketDataType.PROTO, + '::std::vector<::mediapipe::JointList>': + PacketDataType.PROTO_LIST, }