From 1f6c3f3de87e3835a3802e3a62d729211d22a6e4 Mon Sep 17 00:00:00 2001 From: hjwang <2392948297@qq.com> Date: Wed, 10 Jun 2026 11:26:39 +0800 Subject: [PATCH] =?UTF-8?q?refactor(vision):=20=E4=BC=98=E5=8C=96=E5=A7=BF?= =?UTF-8?q?=E6=80=81=E5=85=B3=E9=94=AE=E7=82=B9=E6=A3=80=E6=B5=8B=E5=99=A8?= =?UTF-8?q?=E7=9A=84=E5=88=9D=E5=A7=8B=E5=8C=96=E9=80=BB=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 移除未使用的 threading 和 time 模块导入 - 统一委托类型的使用,避免硬编码委托类型 - 简化 GPU 和 CPU 委托的创建流程 - 修复委托类型传递的一致性问题 --- app/vision/pose_landmarker.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) 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):