From bf6c8a0b63c747e2b99c0e59774c609b799e2277 Mon Sep 17 00:00:00 2001 From: Jiuqiang Tang Date: Sat, 12 Nov 2022 04:33:33 -0800 Subject: [PATCH] Expose ImageEmbedder APIs in PyPI packages. PiperOrigin-RevId: 488033833 --- mediapipe/tasks/python/vision/__init__.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/mediapipe/tasks/python/vision/__init__.py b/mediapipe/tasks/python/vision/__init__.py index 17593357b..cf19813c1 100644 --- a/mediapipe/tasks/python/vision/__init__.py +++ b/mediapipe/tasks/python/vision/__init__.py @@ -18,6 +18,7 @@ import mediapipe.tasks.python.vision.core 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.object_detector @@ -29,6 +30,10 @@ HandLandmarkerOptions = hand_landmarker.HandLandmarkerOptions HandLandmarkerResult = hand_landmarker.HandLandmarkerResult ImageClassifier = image_classifier.ImageClassifier ImageClassifierOptions = image_classifier.ImageClassifierOptions +ImageClassifierResult = image_classifier.ImageClassifierResult +ImageEmbedder = image_embedder.ImageEmbedder +ImageEmbedderOptions = image_embedder.ImageEmbedderOptions +ImageEmbedderResult = image_embedder.ImageEmbedderResult ImageSegmenter = image_segmenter.ImageSegmenter ImageSegmenterOptions = image_segmenter.ImageSegmenterOptions ObjectDetector = object_detector.ObjectDetector @@ -40,6 +45,7 @@ del core del gesture_recognizer del hand_landmarker del image_classifier +del image_embedder del image_segmenter del object_detector del mediapipe