Spaces:
Sleeping
Sleeping
prod
Browse files- api/routers/camera_stream.py +1 -1
- services/pipeline.py +13 -0
api/routers/camera_stream.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
from
|
| 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
|