Expose FaceAligner and LanguageDetector to be public MediaPipe Tasks Python API.
PiperOrigin-RevId: 529227382
This commit is contained in:
parent
b350f72394
commit
c6e3f08282
|
@ -14,9 +14,13 @@
|
|||
|
||||
"""MediaPipe Tasks Text API."""
|
||||
|
||||
import mediapipe.tasks.python.text.language_detector
|
||||
import mediapipe.tasks.python.text.text_classifier
|
||||
import mediapipe.tasks.python.text.text_embedder
|
||||
|
||||
LanguageDetector = language_detector.LanguageDetector
|
||||
LanguageDetectorOptions = language_detector.LanguageDetectorOptions
|
||||
LanguageDetectorResult = language_detector.LanguageDetectorResult
|
||||
TextClassifier = text_classifier.TextClassifier
|
||||
TextClassifierOptions = text_classifier.TextClassifierOptions
|
||||
TextClassifierResult = text_classifier.TextClassifierResult
|
||||
|
@ -26,5 +30,6 @@ TextEmbedderResult = text_embedder.TextEmbedderResult
|
|||
|
||||
# Remove unnecessary modules to avoid duplication in API docs.
|
||||
del mediapipe
|
||||
del language_detector
|
||||
del text_classifier
|
||||
del text_embedder
|
||||
|
|
|
@ -15,6 +15,7 @@
|
|||
"""MediaPipe Tasks Vision API."""
|
||||
|
||||
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_landmarker
|
||||
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.pose_landmarker
|
||||
|
||||
FaceAligner = face_aligner.FaceAligner
|
||||
FaceAlignerOptions = face_aligner.FaceAlignerOptions
|
||||
FaceDetector = face_detector.FaceDetector
|
||||
FaceDetectorOptions = face_detector.FaceDetectorOptions
|
||||
FaceDetectorResult = face_detector.FaceDetectorResult
|
||||
|
@ -65,6 +68,7 @@ RunningMode = core.vision_task_running_mode.VisionTaskRunningMode
|
|||
|
||||
# Remove unnecessary modules to avoid duplication in API docs.
|
||||
del core
|
||||
del face_aligner
|
||||
del face_detector
|
||||
del face_landmarker
|
||||
del face_stylizer
|
||||
|
|
Loading…
Reference in New Issue
Block a user