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

Expose FaceAligner and LanguageDetector to be public MediaPipe Tasks Python API.

Browse Source 
PiperOrigin-RevId: 529227382
This commit is contained in:
Jiuqiang Tang 2023-05-03 16:36:45 -07:00 committed by Copybara-Service
parent b350f72394
commit c6e3f08282
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mediapipe/tasks/python/text/__init__.py
View File

@ -14,9 +14,13 @@
"""MediaPipe Tasks Text API.""" """MediaPipe Tasks Text API."""
import mediapipe.tasks.python.text.language_detector
import mediapipe.tasks.python.text.text_classifier import mediapipe.tasks.python.text.text_classifier
import mediapipe.tasks.python.text.text_embedder import mediapipe.tasks.python.text.text_embedder
LanguageDetector = language_detector.LanguageDetector
LanguageDetectorOptions = language_detector.LanguageDetectorOptions
LanguageDetectorResult = language_detector.LanguageDetectorResult
TextClassifier = text_classifier.TextClassifier TextClassifier = text_classifier.TextClassifier
TextClassifierOptions = text_classifier.TextClassifierOptions TextClassifierOptions = text_classifier.TextClassifierOptions
TextClassifierResult = text_classifier.TextClassifierResult TextClassifierResult = text_classifier.TextClassifierResult
@ -26,5 +30,6 @@ TextEmbedderResult = text_embedder.TextEmbedderResult
# Remove unnecessary modules to avoid duplication in API docs. # Remove unnecessary modules to avoid duplication in API docs.
del mediapipe del mediapipe
del language_detector
del text_classifier del text_classifier
del text_embedder del text_embedder

4
mediapipe/tasks/python/vision/__init__.py
View File

@ -15,6 +15,7 @@
"""MediaPipe Tasks Vision API.""" """MediaPipe Tasks Vision API."""
import mediapipe.tasks.python.vision.core import mediapipe.tasks.python.vision.core
import mediapipe.tasks.python.vision.face_aligner
import mediapipe.tasks.python.vision.face_detector import mediapipe.tasks.python.vision.face_detector
import mediapipe.tasks.python.vision.face_landmarker import mediapipe.tasks.python.vision.face_landmarker
import mediapipe.tasks.python.vision.face_stylizer import mediapipe.tasks.python.vision.face_stylizer
@ -27,6 +28,8 @@ import mediapipe.tasks.python.vision.interactive_segmenter
import mediapipe.tasks.python.vision.object_detector import mediapipe.tasks.python.vision.object_detector
import mediapipe.tasks.python.vision.pose_landmarker import mediapipe.tasks.python.vision.pose_landmarker
FaceAligner = face_aligner.FaceAligner
FaceAlignerOptions = face_aligner.FaceAlignerOptions
FaceDetector = face_detector.FaceDetector FaceDetector = face_detector.FaceDetector
FaceDetectorOptions = face_detector.FaceDetectorOptions FaceDetectorOptions = face_detector.FaceDetectorOptions
FaceDetectorResult = face_detector.FaceDetectorResult FaceDetectorResult = face_detector.FaceDetectorResult
@ -65,6 +68,7 @@ RunningMode = core.vision_task_running_mode.VisionTaskRunningMode
# Remove unnecessary modules to avoid duplication in API docs. # Remove unnecessary modules to avoid duplication in API docs.
del core del core
del face_aligner
del face_detector del face_detector
del face_landmarker del face_landmarker
del face_stylizer del face_stylizer