Registering FaceGeometry proto.
PiperOrigin-RevId: 516663848
This commit is contained in:
parent
f60fe739d3
commit
cd2cc971bb
|
@ -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_frame
|
||||
from mediapipe.python._framework_bindings import packet
|
||||
from mediapipe.tasks.cc.vision.face_geometry.proto import face_geometry_pb2
|
||||
|
||||
CalculatorGraph = calculator_graph.CalculatorGraph
|
||||
Image = image.Image
|
||||
|
@ -178,11 +177,6 @@ class PacketTest(absltest.TestCase):
|
|||
text_format.Parse('score: 0.5', detection)
|
||||
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):
|
||||
p = packet_creator.create_string('abc').at(100)
|
||||
self.assertEqual(packet_getter.get_str(p), 'abc')
|
||||
|
|
Loading…
Reference in New Issue
Block a user