feat(demo): add realtime visualization pipeline flow

Integrate an opt-in OpenCV visualizer into the demo runtime so operators can monitor tracking, segmentation, and inference confidence in real time without changing the default non-visual execution path.
2026-02-27 20:14:24 +08:00
parent 846549498c
commit 4cc2ef7c63
3 changed files with 670 additions and 11 deletions
@@ -1,7 +1,122 @@
 from __future__ import annotations

-from .pipeline import main
+import argparse
+import inspect
+import logging
+import sys
+
+from .pipeline import ScoliosisPipeline


 if __name__ == "__main__":
-    main()
+    parser = argparse.ArgumentParser(description="Scoliosis Detection Pipeline")
+    parser.add_argument(
+        "--source", type=str, required=True, help="Video source path or camera ID"
+    )
+    parser.add_argument(
+        "--checkpoint", type=str, required=True, help="Model checkpoint path"
+    )
+    parser.add_argument(
+        "--config",
+        type=str,
+        default="configs/sconet/sconet_scoliosis1k.yaml",
+        help="Config file path",
+    )
+    parser.add_argument("--device", type=str, default="cuda:0", help="Device to run on")
+    parser.add_argument(
+        "--yolo-model", type=str, default="ckpt/yolo11n-seg.pt", help="YOLO model name"
+    )
+    parser.add_argument(
+        "--window", type=int, default=30, help="Window size for classification"
+    )
+    parser.add_argument("--stride", type=int, default=30, help="Stride for window")
+    parser.add_argument(
+        "--nats-url", type=str, default=None, help="NATS URL for result publishing"
+    )
+    parser.add_argument(
+        "--nats-subject", type=str, default="scoliosis.result", help="NATS subject"
+    )
+    parser.add_argument(
+        "--max-frames", type=int, default=None, help="Maximum frames to process"
+    )
+    parser.add_argument(
+        "--preprocess-only", action="store_true", help="Only preprocess silhouettes"
+    )
+    parser.add_argument(
+        "--silhouette-export-path",
+        type=str,
+        default=None,
+        help="Path to export silhouettes",
+    )
+    parser.add_argument(
+        "--silhouette-export-format", type=str, default="pickle", help="Export format"
+    )
+    parser.add_argument(
+        "--silhouette-visualize-dir",
+        type=str,
+        default=None,
+        help="Directory for silhouette visualizations",
+    )
+    parser.add_argument(
+        "--result-export-path", type=str, default=None, help="Path to export results"
+    )
+    parser.add_argument(
+        "--result-export-format", type=str, default="json", help="Result export format"
+    )
+    parser.add_argument(
+        "--visualize", action="store_true", help="Enable real-time visualization"
+    )
+    args = parser.parse_args()
+
+    logging.basicConfig(
+        level=logging.INFO,
+        format="%(asctime)s %(levelname)s %(name)s: %(message)s",
+    )
+
+    # Validate preprocess-only mode requires silhouette export path
+    if args.preprocess_only and not args.silhouette_export_path:
+        print(
+            "Error: --silhouette-export-path is required when using --preprocess-only",
+            file=sys.stderr,
+        )
+        raise SystemExit(2)
+
+    try:
+        # Import here to avoid circular imports
+        from .pipeline import validate_runtime_inputs
+
+        validate_runtime_inputs(
+            source=args.source, checkpoint=args.checkpoint, config=args.config
+        )
+
+        # Build kwargs based on what ScoliosisPipeline accepts
+        sig = inspect.signature(ScoliosisPipeline.__init__)
+        pipeline_kwargs = {
+            "source": args.source,
+            "checkpoint": args.checkpoint,
+            "config": args.config,
+            "device": args.device,
+            "yolo_model": args.yolo_model,
+            "window": args.window,
+            "stride": args.stride,
+            "nats_url": args.nats_url,
+            "nats_subject": args.nats_subject,
+            "max_frames": args.max_frames,
+            "preprocess_only": args.preprocess_only,
+            "silhouette_export_path": args.silhouette_export_path,
+            "silhouette_export_format": args.silhouette_export_format,
+            "silhouette_visualize_dir": args.silhouette_visualize_dir,
+            "result_export_path": args.result_export_path,
+            "result_export_format": args.result_export_format,
+        }
+        if "visualize" in sig.parameters:
+            pipeline_kwargs["visualize"] = args.visualize
+
+        pipeline = ScoliosisPipeline(**pipeline_kwargs)
+        raise SystemExit(pipeline.run())
+    except ValueError as err:
+        print(f"Error: {err}", file=sys.stderr)
+        raise SystemExit(2) from err
+    except RuntimeError as err:
+        print(f"Runtime error: {err}", file=sys.stderr)
+        raise SystemExit(1) from err