operating-room-monitor-server/app/baked/algorithm.py

"""YOLO 耗材/手部与撕段四模型：路径与超参写死，与业务线部署解耦。"""

from __future__ import annotations

from pathlib import Path

_PACKAGE_DIR = Path(__file__).resolve().parent.parent


def default_consumable_classifier_weights_path() -> str:
    return str(_PACKAGE_DIR / "resources" / "consumable_classifier.pt")


def default_consumable_classifier_labels_yaml_path() -> str:
    return str(_PACKAGE_DIR / "resources" / "consumable_classifier_labels.yaml")


def default_or_site_config_sample_path() -> str:
    """站点配置示例：video_rtsp_urls + voice_or_room_bindings。"""
    return str(_PACKAGE_DIR / "resources" / "or_site_config.sample.json")


# --- 耗材分类（YOLO-cls）---
CONSUMABLE_CLASSIFIER_WEIGHTS: str = default_consumable_classifier_weights_path()
CONSUMABLE_CLASSIFIER_IMGSZ: int = 224
CONSUMABLE_CLASSIFIER_DEVICE: str = ""
CONSUMABLE_CLASSIFIER_TOPK: int = 5
CONSUMABLE_MIN_CLS_CONFIDENCE: float = 0.5
CONSUMABLE_CLASSIFIER_LABELS_YAML_PATH: str = default_consumable_classifier_labels_yaml_path()
CONSUMABLE_VISION_WINDOW_SEC: float = 15.0

# --- 手部检测；全空路径则整帧分类 ---
HAND_DETECTION_WEIGHTS: str = ""
HAND_DETECTION_IMGSZ: int = 640
HAND_DETECTION_CONF: float = 0.25
HAND_DETECTION_PAD_RATIO: float = 0.30
HAND_DETECTION_MIN_CROP_PX: int = 64
HAND_DETECTION_DEVICE: str = ""

# --- 撕段四模型（与 haocai_consumption demo 同构）；无权重文件时勿开启 TEAR_SEGMENT_ENABLED ---
TEAR_SEGMENT_ENABLED: bool = False
TEAR_SEGMENT_PRIMARY_CAMERA_ID: str = ""
TEAR_SEGMENT_HAND_DET_WEIGHTS: str = ""
TEAR_SEGMENT_TEAR_WEIGHTS: str = ""
TEAR_SEGMENT_GOODBAD_WEIGHTS: str = ""
TEAR_SEGMENT_HAOCAI_WEIGHTS: str = ""
TEAR_SEGMENT_LABELS_YAML_PATH: str = ""
TEAR_SEGMENT_ASSUMED_FPS: float = 25.0
TEAR_SEGMENT_DET_CONF: float = 0.25
TEAR_SEGMENT_PAD_RATIO: float = 0.30
TEAR_SEGMENT_TEAR_CONF: float = 0.35
TEAR_SEGMENT_TEAR_SMOOTH: int = 5
TEAR_SEGMENT_GAP_RATIO: float = 1.5
TEAR_SEGMENT_MIN_TEAR_SEC: float = 1.2
TEAR_SEGMENT_MIN_GAP_SEC: float = 1.0
TEAR_SEGMENT_DET_IMGSZ: int = 640
TEAR_SEGMENT_TEAR_IMGSZ: int = 224
TEAR_SEGMENT_GOODBAD_IMGSZ: int = 224
TEAR_SEGMENT_HAOCAI_IMGSZ: int = 224
TEAR_SEGMENT_TEAR_DEVICE: str = ""
TEAR_SEGMENT_GOODBAD_DEVICE: str = ""
TEAR_SEGMENT_HAOCAI_DEVICE: str = ""
TEAR_SEGMENT_LOG_TXT: bool = False
TEAR_SEGMENT_LOG_TXT_PATH: str = "logs/tear_segment_{surgery_id}.txt"