76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
#!/usr/bin/env python3
|
||
"""Debug 入口：Excel I 列时间段替代 ActionFormer，其余 Phase2 与 main.py 一致。"""
|
||
from __future__ import annotations
|
||
|
||
import argparse
|
||
import os
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
PACK_ROOT = Path(__file__).resolve().parent
|
||
sys.path.insert(0, str(PACK_ROOT / "src"))
|
||
|
||
from paths import ensure_code_on_path
|
||
|
||
ensure_code_on_path(PACK_ROOT)
|
||
|
||
from config import load_run_config
|
||
from orchestrator import run_debug_pipeline
|
||
|
||
|
||
def main() -> int:
|
||
os.environ.setdefault("OPENCV_FFMPEG_LOGLEVEL", "8")
|
||
ap = argparse.ArgumentParser(
|
||
description="手术室耗材 Debug 主流程（Excel 时间段 → Phase2，跳过 ActionFormer）"
|
||
)
|
||
ap.add_argument("--video", type=Path, required=True, help="输入 MP4")
|
||
ap.add_argument(
|
||
"--excel",
|
||
type=Path,
|
||
required=True,
|
||
help="商品表 Excel（C 列白名单 + I 列时间段 + 产品编码）",
|
||
)
|
||
ap.add_argument("--out", type=Path, required=True, help="输出 TSV")
|
||
ap.add_argument(
|
||
"--config",
|
||
type=Path,
|
||
default=PACK_ROOT / "configs" / "default_config.yaml",
|
||
help="继承 weights / phase2 / tear_merge / doctor 的 YAML（忽略 io 与 phase1）",
|
||
)
|
||
ap.add_argument(
|
||
"--time-col-index",
|
||
type=int,
|
||
default=8,
|
||
help="时间段列索引，默认 8 即 Excel I 列；视频2 可用 9（J 列）",
|
||
)
|
||
ap.add_argument(
|
||
"--min-seg-seconds",
|
||
type=float,
|
||
default=None,
|
||
help="最短段时长（秒）；默认 0 表示不过滤短段",
|
||
)
|
||
args = ap.parse_args()
|
||
|
||
cfg_path = args.config.resolve()
|
||
if not cfg_path.is_file():
|
||
print("找不到配置:", cfg_path, file=sys.stderr)
|
||
return 1
|
||
|
||
run_cfg = load_run_config(PACK_ROOT, cfg_path)
|
||
run_cfg.video = args.video.resolve()
|
||
run_cfg.excel = args.excel.resolve()
|
||
run_cfg.out = args.out.resolve()
|
||
run_cfg.excel_time_col_index = int(args.time_col_index)
|
||
if args.min_seg_seconds is not None:
|
||
run_cfg.af_min_seg_seconds = float(args.min_seg_seconds)
|
||
else:
|
||
run_cfg.af_min_seg_seconds = 0.0
|
||
|
||
run_cfg.merge_adjacent_tear = False
|
||
|
||
return int(run_debug_pipeline(run_cfg))
|
||
|
||
|
||
if __name__ == "__main__":
|
||
raise SystemExit(main())
|