From e90e53ffaf1aeef3037a6f336b48a82e0b866a4c Mon Sep 17 00:00:00 2001 From: crosstyan Date: Fri, 27 Feb 2026 20:14:35 +0800 Subject: [PATCH] chore(demo): relocate yolo model default under ckpt Point demo defaults and integration tests to the checkpoint directory so runtime configuration matches the canonical model artifact location. --- tests/demo/test_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/demo/test_pipeline.py b/tests/demo/test_pipeline.py index d8c8142..23ab0fa 100644 --- a/tests/demo/test_pipeline.py +++ b/tests/demo/test_pipeline.py @@ -31,7 +31,7 @@ REPO_ROOT: Final[Path] = Path(__file__).resolve().parents[2] SAMPLE_VIDEO_PATH: Final[Path] = REPO_ROOT / "assets" / "sample.mp4" CHECKPOINT_PATH: Final[Path] = REPO_ROOT / "ckpt" / "ScoNet-20000.pt" CONFIG_PATH: Final[Path] = REPO_ROOT / "configs" / "sconet" / "sconet_scoliosis1k.yaml" -YOLO_MODEL_PATH: Final[Path] = REPO_ROOT / "yolo11n-seg.pt" +YOLO_MODEL_PATH: Final[Path] = REPO_ROOT / "ckpt" / "yolo11n-seg.pt" def _device_for_runtime() -> str: