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

No public description

Browse Source 
PiperOrigin-RevId: 568927297
This commit is contained in:
MediaPipe Team 2023-09-27 12:26:06 -07:00 committed by Copybara-Service
parent 61ce228576
commit 787371cfba
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mediapipe/python/BUILD
View File

@ -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",

7
mediapipe/python/solution_base.py
View File

@ -40,6 +40,7 @@ from mediapipe.calculators.util import landmarks_smoothing_calculator_pb2
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 @@ NAME_TO_TYPE: Mapping[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,
}