Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
37 changes: 19 additions & 18 deletions mediapipe/tasks/python/vision/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,20 +14,21 @@

"""MediaPipe Tasks Vision API."""

import mediapipe.tasks.python.vision.core
import mediapipe.tasks.python.vision.drawing_styles
import mediapipe.tasks.python.vision.drawing_utils
import mediapipe.tasks.python.vision.face_detector
import mediapipe.tasks.python.vision.face_landmarker
import mediapipe.tasks.python.vision.gesture_recognizer
import mediapipe.tasks.python.vision.hand_landmarker
import mediapipe.tasks.python.vision.holistic_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
import mediapipe.tasks.python.vision.pose_landmarker
from mediapipe.tasks.python.vision import drawing_styles
from mediapipe.tasks.python.vision import drawing_utils
from mediapipe.tasks.python.vision import face_detector
from mediapipe.tasks.python.vision import face_landmarker
from mediapipe.tasks.python.vision import gesture_recognizer
from mediapipe.tasks.python.vision import hand_landmarker
from mediapipe.tasks.python.vision import holistic_landmarker
from mediapipe.tasks.python.vision import image_classifier
from mediapipe.tasks.python.vision import image_embedder
from mediapipe.tasks.python.vision import image_segmenter
from mediapipe.tasks.python.vision import interactive_segmenter
from mediapipe.tasks.python.vision import object_detector
from mediapipe.tasks.python.vision import pose_landmarker
from mediapipe.tasks.python.vision.core import image_processing_options
from mediapipe.tasks.python.vision.core import vision_task_running_mode

drawing_styles = drawing_styles
drawing_utils = drawing_utils
Expand Down Expand Up @@ -56,7 +57,7 @@
ImageEmbedderResult = image_embedder.ImageEmbedderResult
ImageSegmenter = image_segmenter.ImageSegmenter
ImageSegmenterOptions = image_segmenter.ImageSegmenterOptions
ImageProcessingOptions = core.image_processing_options.ImageProcessingOptions
ImageProcessingOptions = image_processing_options.ImageProcessingOptions
InteractiveSegmenter = interactive_segmenter.InteractiveSegmenter
InteractiveSegmenterOptions = interactive_segmenter.InteractiveSegmenterOptions
InteractiveSegmenterRegionOfInterest = interactive_segmenter.RegionOfInterest
Expand All @@ -69,19 +70,19 @@
PoseLandmarkerResult = pose_landmarker.PoseLandmarkerResult
PoseLandmarksConnections = pose_landmarker.PoseLandmarksConnections

RunningMode = core.vision_task_running_mode.VisionTaskRunningMode
RunningMode = vision_task_running_mode.VisionTaskRunningMode

# Remove unnecessary modules to avoid duplication in API docs.
del core
del face_detector
del face_landmarker
del gesture_recognizer
del hand_landmarker
del holistic_landmarker
del image_classifier
del image_embedder
del image_processing_options
del image_segmenter
del interactive_segmenter
del object_detector
del pose_landmarker
del mediapipe
del vision_task_running_mode