diff --git a/src/mediapipe_vision/face_detection/detector.py b/src/mediapipe_vision/face_detection/detector.py index 0e40e6e..44f73f4 100644 --- a/src/mediapipe_vision/face_detection/detector.py +++ b/src/mediapipe_vision/face_detection/detector.py @@ -9,7 +9,7 @@ import torch from mediapipe.tasks import python from mediapipe.tasks.python import vision from mediapipe.tasks.python.core.base_options import BaseOptions -from typing import List, Tuple +from typing import List # Import new types from ..types import FaceDetectionResult, BoundingBox, FaceKeypoint from ...utils.timing import TimestampProvider # Import from utils diff --git a/src/mediapipe_vision/face_landmark/detector.py b/src/mediapipe_vision/face_landmark/detector.py index 73b448d..ca9d165 100644 --- a/src/mediapipe_vision/face_landmark/detector.py +++ b/src/mediapipe_vision/face_landmark/detector.py @@ -6,7 +6,7 @@ from mediapipe.tasks.python import vision from mediapipe.tasks.python.core.base_options import BaseOptions from typing import List, Tuple, Optional, Dict # Import new types -from ..types import LandmarkPoint, FaceLandmarksResult +from ..types import LandmarkPoint from ...utils.timing import TimestampProvider # Import TimestampProvider class FaceLandmarkDetector: diff --git a/src/mediapipe_vision/face_stylization/detector.py b/src/mediapipe_vision/face_stylization/detector.py index 706a0de..cb23d71 100644 --- a/src/mediapipe_vision/face_stylization/detector.py +++ b/src/mediapipe_vision/face_stylization/detector.py @@ -3,7 +3,6 @@ import numpy as np import torch from mediapipe.tasks import python from mediapipe.tasks.python import vision -from typing import List, Optional from ...utils.timing import TimestampProvider class FaceStylizer: diff --git a/src/mediapipe_vision/hand_landmark/detector.py b/src/mediapipe_vision/hand_landmark/detector.py index 693764d..613b3c2 100644 --- a/src/mediapipe_vision/hand_landmark/detector.py +++ b/src/mediapipe_vision/hand_landmark/detector.py @@ -4,7 +4,7 @@ import torch from mediapipe.tasks import python from mediapipe.tasks.python import vision from mediapipe.tasks.python.core.base_options import BaseOptions -from typing import List, Tuple, Optional +from typing import List # Import new types from ..types import LandmarkPoint, HandLandmarksResult from ...utils.timing import TimestampProvider # Import TimestampProvider diff --git a/src/mediapipe_vision/holistic_landmark/detector.py b/src/mediapipe_vision/holistic_landmark/detector.py index 33b157c..5d5bbfb 100644 --- a/src/mediapipe_vision/holistic_landmark/detector.py +++ b/src/mediapipe_vision/holistic_landmark/detector.py @@ -8,7 +8,7 @@ import mediapipe as mp import numpy as np import torch import cv2 -from typing import List, Tuple, Optional, Dict, Any +from typing import List, Tuple, Optional, Any # Import needed types from ..types import LandmarkPoint, HolisticLandmarksResult diff --git a/src/mediapipe_vision/image_embedding/detector.py b/src/mediapipe_vision/image_embedding/detector.py index 41aad69..eb7152c 100644 --- a/src/mediapipe_vision/image_embedding/detector.py +++ b/src/mediapipe_vision/image_embedding/detector.py @@ -4,7 +4,7 @@ import torch from mediapipe.tasks import python from mediapipe.tasks.python import vision from mediapipe.tasks.python.core.base_options import BaseOptions -from typing import List, Optional +from typing import List # Import new types from ..types import ImageEmbedderResult # Import TimestampProvider from the new location diff --git a/src/mediapipe_vision/interactive_segmentation/detector.py b/src/mediapipe_vision/interactive_segmentation/detector.py index c9aafd4..1ce0f1a 100644 --- a/src/mediapipe_vision/interactive_segmentation/detector.py +++ b/src/mediapipe_vision/interactive_segmentation/detector.py @@ -3,10 +3,6 @@ import mediapipe as mp import numpy as np import torch from mediapipe.tasks import python -from mediapipe.tasks.python import vision -from mediapipe.tasks.python.vision.interactive_segmenter import ( - InteractiveSegmenter, -) from mediapipe.tasks.python.vision import interactive_segmenter from mediapipe.tasks.python.components.containers import keypoint