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

Removed dummy packet creation and preserved face_geometry protobuf import

Browse Source
This commit is contained in:
kinaryml 2023-03-15 10:52:16 -07:00
parent 4a6015e65c
commit 80dd764605
1 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mediapipe/tasks/python/vision/face_landmarker.py
View File

@ -25,7 +25,7 @@ from mediapipe.python import packet_getter
from mediapipe.python._framework_bindings import image as image_module from mediapipe.python._framework_bindings import image as image_module
from mediapipe.python._framework_bindings import packet as packet_module from mediapipe.python._framework_bindings import packet as packet_module
from mediapipe.tasks.cc.vision.face_landmarker.proto import face_landmarker_graph_options_pb2 from mediapipe.tasks.cc.vision.face_landmarker.proto import face_landmarker_graph_options_pb2
# TODO: Remove later. # TODO: Remove this later.
from mediapipe.tasks.cc.vision.face_geometry.proto import face_geometry_pb2 from mediapipe.tasks.cc.vision.face_geometry.proto import face_geometry_pb2
from mediapipe.tasks.python.components.containers import category as category_module from mediapipe.tasks.python.components.containers import category as category_module
from mediapipe.tasks.python.components.containers import landmark as landmark_module from mediapipe.tasks.python.components.containers import landmark as landmark_module
@ -162,12 +162,6 @@ def _build_landmarker_result(
category_name=face_blendshapes.label)) category_name=face_blendshapes.label))
face_blendshapes_results.append(face_blendshapes_categories) face_blendshapes_results.append(face_blendshapes_categories)
# Creates a dummy FaceGeometry packet to initialize the symbol database.
# TODO: Remove later.
face_geometry_in = face_geometry_pb2.FaceGeometry()
p = packet_creator.create_proto(face_geometry_in).at(100)
face_geometry_out = packet_getter.get_proto(p)
facial_transformation_matrixes_results = [] facial_transformation_matrixes_results = []
if _FACE_GEOMETRY_STREAM_NAME in output_packets: if _FACE_GEOMETRY_STREAM_NAME in output_packets:
facial_transformation_matrixes_proto_list = packet_getter.get_proto_list( facial_transformation_matrixes_proto_list = packet_getter.get_proto_list(