diff --git a/app/vision/pose_landmarker.py b/app/vision/pose_landmarker.py index 13de5e9..c831b69 100644 --- a/app/vision/pose_landmarker.py +++ b/app/vision/pose_landmarker.py @@ -1,7 +1,5 @@ from __future__ import annotations -import threading -import time from typing import Callable import mediapipe as mp @@ -30,14 +28,14 @@ class PoseLandmarkerWrapper: if prefer_gpu: try: self.delegate = BaseOptions.Delegate.GPU - self._landmarker = self._create(PoseLandmarker.Delegate.GPU) + self._landmarker = self._create(self.delegate, result_callback) logger.info("MediaPipe PoseLandmarker initialized with GPU delegate") return except Exception as exc: logger.warning("MediaPipe GPU delegate unavailable, falling back to CPU: {}", exc) self.delegate = BaseOptions.Delegate.CPU - self._landmarker = self._create(PoseLandmarker.Delegate.CPU, result_callback) + self._landmarker = self._create(self.delegate, result_callback) logger.info("MediaPipe PoseLandmarker initialized with CPU delegate") def _create(self, delegate, result_callback=None):