phase cv 0

detect/graph.py

@@ -17,6 +17,7 @@ from detect.profiles import SoccerBroadcastProfile
 from detect.state import DetectState
 from detect.stages.frame_extractor import extract_frames
 from detect.stages.scene_filter import scene_filter
+from detect.stages.edge_detector import detect_edge_regions
 from detect.stages.yolo_detector import detect_objects
 from detect.stages.preprocess import preprocess_regions
 from detect.stages.ocr_stage import run_ocr
@@ -31,6 +32,7 @@ INFERENCE_URL = os.environ.get("INFERENCE_URL")  # None = local mode
 NODES = [
     "extract_frames",
     "filter_scenes",
+    "detect_edges",
     "detect_objects",
     "preprocess",
     "run_ocr",
@@ -119,6 +121,28 @@ def node_filter_scenes(state: DetectState) -> dict:
     return {"filtered_frames": kept, "stats": stats}
 
 
+def node_detect_edges(state: DetectState) -> dict:
+    _emit_transition(state, "detect_edges", "running")
+
+    with trace_node(state, "detect_edges") as span:
+        profile = _get_profile(state)
+        config = profile.region_analysis_config()
+        frames = state.get("filtered_frames", [])
+        job_id = state.get("job_id")
+
+        regions = detect_edge_regions(
+            frames, config, inference_url=INFERENCE_URL, job_id=job_id,
+        )
+        total = sum(len(r) for r in regions.values())
+        span.set_output({"frames": len(frames), "edge_regions": total})
+
+    stats = state.get("stats", PipelineStats())
+    stats.cv_regions_detected = total
+
+    _emit_transition(state, "detect_edges", "done")
+    return {"edge_regions_by_frame": regions, "stats": stats}
+
+
 def node_detect_objects(state: DetectState) -> dict:
     _emit_transition(state, "detect_objects", "running")
 
@@ -359,6 +383,7 @@ def _checkpointing_node(node_name: str, node_fn):
 NODE_FUNCTIONS = [
     ("extract_frames", node_extract_frames),
     ("filter_scenes", node_filter_scenes),
+    ("detect_edges", node_detect_edges),
     ("detect_objects", node_detect_objects),
     ("preprocess", node_preprocess),
     ("run_ocr", node_run_ocr),