diff --git a/node_wrappers/mediapipe_vision/common/base_detector_node.py b/node_wrappers/mediapipe_vision/common/base_detector_node.py index 662c681..80b2097 100644 --- a/node_wrappers/mediapipe_vision/common/base_detector_node.py +++ b/node_wrappers/mediapipe_vision/common/base_detector_node.py @@ -1,8 +1,7 @@ """Base node wrapper for MediaPipe Vision detectors.""" -import torch import logging -from typing import Any, Dict, List, Tuple, Optional, Type, Callable, Union +from typing import Dict logger = logging.getLogger(__name__) diff --git a/node_wrappers/mediapipe_vision/common/base_visualization_nodes.py b/node_wrappers/mediapipe_vision/common/base_visualization_nodes.py index f258cd0..7470124 100644 --- a/node_wrappers/mediapipe_vision/common/base_visualization_nodes.py +++ b/node_wrappers/mediapipe_vision/common/base_visualization_nodes.py @@ -1,7 +1,6 @@ import logging -from typing import List, Dict, Tuple, Any, Optional, Type +from typing import List, Any, Optional import cv2 -import numpy as np from ....src.utils.image import convert_to_cv2, convert_to_tensor diff --git a/node_wrappers/mediapipe_vision/common/model_loader.py b/node_wrappers/mediapipe_vision/common/model_loader.py index 76ae1b6..36f3ec0 100644 --- a/node_wrappers/mediapipe_vision/common/model_loader.py +++ b/node_wrappers/mediapipe_vision/common/model_loader.py @@ -6,7 +6,7 @@ import logging import traceback # For better error logging # Add import for MEDIAPIPE_MODELS dictionary and the get_model_description function -from ....src.mediapipe_vision.model_loader import MEDIAPIPE_MODELS, get_model_description +from ....src.mediapipe_vision.model_loader import get_model_description logger = logging.getLogger(__name__)