From d8a2fc44f99e3e95c99015bd74102970b491b159 Mon Sep 17 00:00:00 2001 From: Kinar Date: Thu, 14 Sep 2023 06:17:45 +0530 Subject: [PATCH] Removed running mode from Face Stylizer --- mediapipe/tasks/python/test/vision/BUILD | 1 - mediapipe/tasks/python/test/vision/face_stylizer_test.py | 2 -- mediapipe/tasks/python/vision/face_stylizer.py | 5 ----- 3 files changed, 8 deletions(-) diff --git a/mediapipe/tasks/python/test/vision/BUILD b/mediapipe/tasks/python/test/vision/BUILD index 632cc11d4..c6fae0e6c 100644 --- a/mediapipe/tasks/python/test/vision/BUILD +++ b/mediapipe/tasks/python/test/vision/BUILD @@ -226,6 +226,5 @@ py_test( "//mediapipe/tasks/python/test:test_utils", "//mediapipe/tasks/python/vision:face_stylizer", "//mediapipe/tasks/python/vision/core:image_processing_options", - "//mediapipe/tasks/python/vision/core:vision_task_running_mode", ], ) diff --git a/mediapipe/tasks/python/test/vision/face_stylizer_test.py b/mediapipe/tasks/python/test/vision/face_stylizer_test.py index 4f5392128..2673060cc 100644 --- a/mediapipe/tasks/python/test/vision/face_stylizer_test.py +++ b/mediapipe/tasks/python/test/vision/face_stylizer_test.py @@ -24,7 +24,6 @@ from mediapipe.tasks.python.core import base_options as base_options_module from mediapipe.tasks.python.components.containers import rect from mediapipe.tasks.python.test import test_utils from mediapipe.tasks.python.vision import face_stylizer -from mediapipe.tasks.python.vision.core import vision_task_running_mode as running_mode_module from mediapipe.tasks.python.vision.core import image_processing_options as image_processing_options_module @@ -33,7 +32,6 @@ _Rect = rect.Rect _Image = image_module.Image _FaceStylizer = face_stylizer.FaceStylizer _FaceStylizerOptions = face_stylizer.FaceStylizerOptions -_RUNNING_MODE = running_mode_module.VisionTaskRunningMode _ImageProcessingOptions = image_processing_options_module.ImageProcessingOptions _MODEL = 'face_stylizer_color_ink.task' diff --git a/mediapipe/tasks/python/vision/face_stylizer.py b/mediapipe/tasks/python/vision/face_stylizer.py index 460eee353..af70a61dc 100644 --- a/mediapipe/tasks/python/vision/face_stylizer.py +++ b/mediapipe/tasks/python/vision/face_stylizer.py @@ -31,7 +31,6 @@ _BaseOptions = base_options_module.BaseOptions _FaceStylizerGraphOptionsProto = ( face_stylizer_graph_options_pb2.FaceStylizerGraphOptions ) -_RunningMode = running_mode_module.VisionTaskRunningMode _ImageProcessingOptions = image_processing_options_module.ImageProcessingOptions _TaskInfo = task_info_module.TaskInfo @@ -55,15 +54,11 @@ class FaceStylizerOptions: """ base_options: _BaseOptions - running_mode: _RunningMode = _RunningMode.IMAGE @doc_controls.do_not_generate_docs def to_pb2(self) -> _FaceStylizerGraphOptionsProto: """Generates an FaceStylizerOptions protobuf object.""" base_options_proto = self.base_options.to_pb2() - base_options_proto.use_stream_mode = ( - False if self.running_mode == _RunningMode.IMAGE else True - ) return _FaceStylizerGraphOptionsProto(base_options=base_options_proto)