8a4bad99d3
- 新增 app/baked/algorithm|pipeline，非部署参数不再走 env；Settings 保留 DB/HTTP/RTSP/海康/百度/MinIO/Demo - 移除 init_db_schema 与 reload 配置；main 仅 check_database；start*.sh 在 uvicorn 前执行 alembic upgrade head - 依赖 psycopg[binary] 供 Alembic 同步 URL；alembic/env 注释与预发清单更新 - 撕段门控消费管线、各视频/语音/归档调用改为 baked - 百度环境变量仅 BAIDU_APP_ID、BAIDU_API_KEY、BAIDU_SECRET_KEY 与 BAIDU_* 超时/ASR；人脸脚本与 baidu_speech 文案同步 - 全量单测与 .env.example 更新；.gitignore 忽略 refs/（本地权重/视频不入库） Made-with: Cursor
64 lines
2.3 KiB
Python
64 lines
2.3 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_camera_rtsp_urls_sample_path() -> str:
|
||
return str(_PACKAGE_DIR / "resources" / "camera_rtsp_urls.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"
|