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

Registering FaceGeometry proto.

Browse Source 
PiperOrigin-RevId: 516663848
This commit is contained in:
Jiuqiang Tang 2023-03-14 16:24:56 -07:00 committed by Copybara-Service
parent f60fe739d3
commit cd2cc971bb
1 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mediapipe/python/packet_test.py
View File

@ -28,7 +28,6 @@ from mediapipe.python._framework_bindings import calculator_graph
from mediapipe.python._framework_bindings import image from mediapipe.python._framework_bindings import image
from mediapipe.python._framework_bindings import image_frame from mediapipe.python._framework_bindings import image_frame
from mediapipe.python._framework_bindings import packet from mediapipe.python._framework_bindings import packet
from mediapipe.tasks.cc.vision.face_geometry.proto import face_geometry_pb2
CalculatorGraph = calculator_graph.CalculatorGraph CalculatorGraph = calculator_graph.CalculatorGraph
Image = image.Image Image = image.Image
@ -178,11 +177,6 @@ class PacketTest(absltest.TestCase):
text_format.Parse('score: 0.5', detection) text_format.Parse('score: 0.5', detection)
p = packet_creator.create_proto(detection).at(100) p = packet_creator.create_proto(detection).at(100)
def test_face_geometry_proto_packet(self):
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)
def test_string_packet(self): def test_string_packet(self):
p = packet_creator.create_string('abc').at(100) p = packet_creator.create_string('abc').at(100)
self.assertEqual(packet_getter.get_str(p), 'abc') self.assertEqual(packet_getter.get_str(p), 'abc')