update yolov8 workload in deployment

base/hub/kerberos-hub-yolov8.yaml

@@ -1,31 +1,26 @@
 apiVersion: apps/v1
 kind: Deployment
 metadata:
-  name: data-filtering
+  name: hub-yolov8
   labels:
-    app: data-filtering
+    app: hub-yolov8
 spec:
   replicas: 1
   selector:
     matchLabels:
-      app: data-filtering
+      app: hub-yolov8
   template:
     metadata:
       labels:
-        app: data-filtering
+        app: hub-yolov8
     spec:
       containers:
-        - name: data-filtering
-          image: uugai/data-filter:1.0.4
-          resources:
-            limits:
-              nvidia.com/gpu: 1 # requesting a single GPU
+        - name: hub-yolov8
+          image: uugai/hub-yolov8:latest
+          #resources:
+          #  limits:
+          #    nvidia.com/gpu: 1 # requesting a single GPU
           env:
-            - name: MODEL_NAME
-              value: "yolov8n.pt"
-            - name: CONDITION
-              value:  "1 persons detected"
-
             - name: QUEUE_NAME
               value: "data-filtering" # This is the topic of kafka we will read messages from.
             - name: QUEUE_HOST
@@ -44,16 +39,35 @@ spec:
             - name: STORAGE_SECRET_KEY
               value: "OGGqat4lXRpL@9XBYc8FUaId@5"
 
+
+            # YOLOv8 parameters
+            - name: MODEL_NAME
+              value: "yolov8n.pt"
+            - name: MEDIA_SAVEPATH
+              value: "/ml/data/input/video.mp4"
+
             - name: LOGGING
               value: "True"
             - name: PLOT
               value: "False"
             - name: SAVE_VIDEO
               value: "False"
-            - name: MEDIA_SAVEPATH
-              value: "/ml/data/input/video.mp4"
             - name: OUTPUT_MEDIA_SAVEPATH
-              value: "/ml/data/input/output_video.mp4"
+              value: "path/to/your/output_video.mp4"
+
+            - name: CREATE_BBOX_FRAME
+              value: "False"
+            - name: SAVE_BBOX_FRAME
+              value: "False"
+            - name: BBOX_FRAME_SAVEPATH
+              value: "path/to/your/output_bbox.jpg"
+
+            - name: CREATE_RETURN_JSON
+              value: "True"
+            - name: SAVE_RETURN_JSON
+              value: "False"
+            - name: RETURN_JSON_SAVEPATH
+              value: "path/to/your/output_json.json"
 
             - name: TIME_VERBOSE
               value: "True"
@@ -62,12 +76,21 @@ spec:
               value: "3"
             - name: CLASSIFICATION_THRESHOLD
               value: "0.3"
+            - name: MIN_DETECTIONS
+              value: "5"
+            - name: MIN_DISTANCE
+              value: "150"
+            - name: MIN_STATIC_DISTANCE
+              value: "0"
             - name: MAX_NUMBER_OF_PREDICTIONS
               value: "100"
+            - name: FIND_DOMINANT_COLORS
+              value: "False"
             - name: ALLOWED_CLASSIFICATIONS
               value: "0, 1, 2, 3, 5, 7, 14, 15, 16, 24, 26, 28"
-
-            - name: FORWARDING_MEDIA
-              value: "True"
-            - name: REMOVE_AFTER_PROCESSED
-              value: "True"
+            - name: COLOR_PREDICTION_INTERVAL
+              value: "5"
+            - name: MIN_CLUSTERS
+              value: "4"
+            - name: MAX_CLUSTERS
+              value: "4"

base/hub/kustomization.yaml

@@ -1,4 +1,5 @@
 namespace: kerberos-hub
 resources:
   - kerberos-hub-nodeports.yaml
-  - kerberos-hub-import-database-job.yaml
+  - kerberos-hub-import-database-job.yaml
+  - kerberos-hub-yolov8.yaml