e1250 commited on
Commit
688d2c9
·
1 Parent(s): efc36ce
api/routers/camera_stream.py CHANGED
@@ -1,4 +1,4 @@
1
- from backend.domain.detection_box_center import calculate_detection_box_center
2
  from api.dependencies import get_safety_detection_model
3
  from api.dependencies import get_detection_model, get_depth_model
4
  import asyncio
 
1
+ from domain.detection_box_center import calculate_detection_box_center
2
  from api.dependencies import get_safety_detection_model
3
  from api.dependencies import get_detection_model, get_depth_model
4
  import asyncio
services/pipeline.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ class ProcessingPipeline:
2
+ def __init__(self, detector, depth_model, safety_detector, redis_client):
3
+ self.detector = detector
4
+ self.depth_model = depth_model
5
+ self.safety_detector = safety_detector
6
+ self.redis_client = redis_client
7
+
8
+ async def run(self, camera_id:str, image_array):
9
+
10
+ # Run ai models
11
+ # Domain logic
12
+ # save to infra
13
+ pass