From 8cedb82df13b43f95c491b43f0c0694f9abc817c Mon Sep 17 00:00:00 2001 From: Jiuqiang Tang Date: Fri, 7 Apr 2023 12:10:04 -0700 Subject: [PATCH] Expose face stylizer and interactive segmenter to be public mediapipe python API. PiperOrigin-RevId: 522654239 --- mediapipe/tasks/python/vision/__init__.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/mediapipe/tasks/python/vision/__init__.py b/mediapipe/tasks/python/vision/__init__.py index f75157cf..49fe0305 100644 --- a/mediapipe/tasks/python/vision/__init__.py +++ b/mediapipe/tasks/python/vision/__init__.py @@ -17,11 +17,13 @@ import mediapipe.tasks.python.vision.core import mediapipe.tasks.python.vision.face_detector import mediapipe.tasks.python.vision.face_landmarker +import mediapipe.tasks.python.vision.face_stylizer import mediapipe.tasks.python.vision.gesture_recognizer import mediapipe.tasks.python.vision.hand_landmarker import mediapipe.tasks.python.vision.image_classifier import mediapipe.tasks.python.vision.image_embedder import mediapipe.tasks.python.vision.image_segmenter +import mediapipe.tasks.python.vision.interactive_segmenter import mediapipe.tasks.python.vision.object_detector FaceDetector = face_detector.FaceDetector @@ -30,6 +32,8 @@ FaceDetectorResult = face_detector.FaceDetectorResult FaceLandmarker = face_landmarker.FaceLandmarker FaceLandmarkerOptions = face_landmarker.FaceLandmarkerOptions FaceLandmarkerResult = face_landmarker.FaceLandmarkerResult +FaceStylizer = face_stylizer.FaceStylizer +FaceStylizerOptions = face_stylizer.FaceStylizerOptions GestureRecognizer = gesture_recognizer.GestureRecognizer GestureRecognizerOptions = gesture_recognizer.GestureRecognizerOptions GestureRecognizerResult = gesture_recognizer.GestureRecognizerResult @@ -44,6 +48,9 @@ ImageEmbedderOptions = image_embedder.ImageEmbedderOptions ImageEmbedderResult = image_embedder.ImageEmbedderResult ImageSegmenter = image_segmenter.ImageSegmenter ImageSegmenterOptions = image_segmenter.ImageSegmenterOptions +InteractiveSegmenter = interactive_segmenter.InteractiveSegmenter +InteractiveSegmenterOptions = interactive_segmenter.InteractiveSegmenterOptions +InteractiveSegmenterRegionOfInterest = interactive_segmenter.RegionOfInterest ObjectDetector = object_detector.ObjectDetector ObjectDetectorOptions = object_detector.ObjectDetectorOptions RunningMode = core.vision_task_running_mode.VisionTaskRunningMode @@ -52,10 +59,12 @@ RunningMode = core.vision_task_running_mode.VisionTaskRunningMode del core del face_detector del face_landmarker +del face_stylizer del gesture_recognizer del hand_landmarker del image_classifier del image_embedder del image_segmenter +del interactive_segmenter del object_detector del mediapipe