模型调用

2026-08-17 15:49:01 +08:00
parent 8a83deddd3
commit 42026d43e5
5 changed files with 189 additions and 3 deletions
@@ -162,7 +162,11 @@ def cmd_str():
            and _loc_black == "yolo"
            and bool(getattr(config, "TRIANGLE_BLACK_YOLO_PRELOAD_ON_BOOT", True))
        )
-        _preload_yolo = _preload_yolo or _need_black_preload
+        _need_target_preload = (
+            bool(getattr(config, "TARGET_CLASS_YOLO_ENABLE", False))
+            and bool(getattr(config, "TARGET_CLASS_YOLO_PRELOAD_ON_BOOT", True))
+        )
+        _preload_yolo = _preload_yolo or _need_black_preload or _need_target_preload
        if _preload_yolo:
            preload_yolo_detector(logger)
    except Exception as e: