Send raw image over WebRTC if no visualisation is available

roboflow · Jan 22, 2025 · 891ef4b · 891ef4b
1 parent 8aae13e
commit 891ef4b
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/inference/core/interfaces/stream_manager/manager_app/inference_pipeline_manager.py b/inference/core/interfaces/stream_manager/manager_app/inference_pipeline_manager.py
@@ -275,6 +275,16 @@ def start_loop(loop: asyncio.AbstractEventLoop):
             def webrtc_sink(
                 prediction: Dict[str, WorkflowImageData], video_frame: VideoFrame
             ) -> None:
+                if parsed_payload.stream_output[0] not in prediction:
+                    from_inference_queue.sync_put(
+                        video_frame.image
+                    )
+                    return
+                if prediction[parsed_payload.stream_output[0]] is None:
+                    from_inference_queue.sync_put(
+                        video_frame.image
+                    )
+                    return
                 from_inference_queue.sync_put(
                     prediction[parsed_payload.stream_output[0]].numpy_image
                 )