6b3adb4ad8
- 用 OR_SITE_CONFIG_JSON_FILE 统一术间配置（video_rtsp_urls + voice_or_room_bindings） - VoiceTerminalHub：assignment、WS 推送与 HTTP 查询；开录/停录后 notify - 一键联调 orchestrate-and-start 与 /client/surgeries/start 共用指派逻辑，修复 demo 路径不发 WS - 语音桌面端：SIGINT 退出、shutdown 清理、仅 WS 指派、固定 pending 轮询间隔、界面仅保留录音时长 - 新增/调整契约与绑定测试，文档与示例配置同步 Made-with: Cursor
65 lines
2.4 KiB
Python
65 lines
2.4 KiB
Python
"""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"
|