No public description

PiperOrigin-RevId: 568927297
google-ai-edge · Sep 27, 2023 · 787371c · 787371c
1 parent 61ce228
commit 787371c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git 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
@@ -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,
 }