67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
"""
|
|
yolo_api.py — YOLO detection interface with fallback stubs
|
|
"""
|
|
import sys
|
|
import os
|
|
from Core.config_loader import load_config
|
|
from Core.env_loader import PROJECT_ROOT
|
|
|
|
YOLO_AVAILABLE = False
|
|
|
|
# Fallback stubs
|
|
def _stub_sees(c, **kw): return False
|
|
def _stub_count(c): return 0
|
|
def _stub_closest(c="person"): return None
|
|
def _stub_summary(): return "YOLO not loaded"
|
|
def _stub_ppe(): return []
|
|
def _stub_too_close(**k): return False
|
|
def _stub_all(): return set()
|
|
def _stub_fps(): return 0.0
|
|
|
|
yolo_sees = _stub_sees
|
|
yolo_count = _stub_count
|
|
yolo_closest = _stub_closest
|
|
yolo_summary = _stub_summary
|
|
yolo_ppe_violations = _stub_ppe
|
|
yolo_person_too_close = _stub_too_close
|
|
yolo_all_classes = _stub_all
|
|
yolo_fps = _stub_fps
|
|
|
|
|
|
def init_yolo(raw_frame_ref, frame_lock) -> bool:
|
|
"""Start YOLO inference. Returns True if successful."""
|
|
global YOLO_AVAILABLE
|
|
global yolo_sees, yolo_count, yolo_closest, yolo_summary
|
|
global yolo_ppe_violations, yolo_person_too_close, yolo_all_classes, yolo_fps
|
|
|
|
# marcus_yolo.py lives in Vision/
|
|
models_dir = os.path.join(PROJECT_ROOT, "Vision")
|
|
if models_dir not in sys.path:
|
|
sys.path.insert(0, models_dir)
|
|
|
|
try:
|
|
from marcus_yolo import (
|
|
start_yolo, yolo_sees as _ys, yolo_count as _yc, yolo_closest as _ycl,
|
|
yolo_summary as _ysu, yolo_ppe_violations as _ypp,
|
|
yolo_person_too_close as _yptc, yolo_all_classes as _yac, yolo_fps as _yfps,
|
|
)
|
|
except ImportError as e:
|
|
print(f"marcus_yolo.py not found ({e})")
|
|
return False
|
|
|
|
# GPU is required — let RuntimeError from _resolve_device propagate so
|
|
# Marcus hard-fails at startup instead of silently running without vision.
|
|
ok = start_yolo(raw_frame_ref=raw_frame_ref, frame_lock=frame_lock)
|
|
if ok:
|
|
YOLO_AVAILABLE = True
|
|
yolo_sees = _ys
|
|
yolo_count = _yc
|
|
yolo_closest = _ycl
|
|
yolo_summary = _ysu
|
|
yolo_ppe_violations = _ypp
|
|
yolo_person_too_close = _yptc
|
|
yolo_all_classes = _yac
|
|
yolo_fps = _yfps
|
|
print(f"YOLO {'started' if ok else 'failed to start'}")
|
|
return ok
|