From aa02253ebcdeb076462b3e02dcaa95be4a50053b Mon Sep 17 00:00:00 2001
From: Cedric Verstraeten <cedric@verstraeten.io>
Date: Sat, 10 Aug 2024 20:04:20 +0200
Subject: [PATCH] Update data-filtering-deployment.yaml

---
 data-filtering-deployment.yaml | 58 +++++++---------------------------
 1 file changed, 11 insertions(+), 47 deletions(-)

diff --git a/data-filtering-deployment.yaml b/data-filtering-deployment.yaml
index 956d7e8..d05679e 100644
--- a/data-filtering-deployment.yaml
+++ b/data-filtering-deployment.yaml
@@ -21,6 +21,9 @@ spec:
             limits:
               nvidia.com/gpu: 1 # requesting a single GPU
           env:
+            - name: MODEL_NAME
+              value: "yolov8n.pt"
+
             - name: QUEUE_NAME
               value: "data-filtering" # This is the topic of kafka we will read messages from.
             - name: QUEUE_HOST
@@ -38,67 +41,28 @@ spec:
             - name: STORAGE_SECRET_KEY
               value: "k8DrcB@hQ5XfxDENzDKcnkxBHx"
 
-            - 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: "True"
+            - name: MEDIA_SAVEPATH
+              value: "/ml/data/input/video.mp4"
             - name: OUTPUT_MEDIA_SAVEPATH
               value: "/ml/data/input/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"
 
             - name: CLASSIFICATION_FPS
               value: "3"
-            - name: CLASSIFICATION_THRESHOLD
-              value: "0.3"
-            - name: MIN_DETECTIONS
-              value: "5"
-            - name: MIN_DISTANCE
-              value: "100"
-            - name: MIN_STATIC_DISTANCE
-              value: "50"
-            - name: MAX_NUMBER_OF_PREDICTIONS
-              value: "100"
-            - name: FIND_DOMINANT_COLORS
-              value: "True"
             - name: ALLOWED_CLASSIFICATIONS
               value: "0, 1, 2, 3, 5, 7, 14, 15, 16, 24, 26, 28"
-            - name: COLOR_PREDICTION_INTERVAL
-              value: "5"
-            - name: MIN_CLUSTERS
-              value: "4"
-            - name: MAX_CLUSTERS
-              value: "4"
+            - name: CLASSIFICATION_THRESHOLD
+              value: "0.3"
 
-            #- name: FORWARDING_MEDIA
-            #  value: "false"
-            #- name: FORWARDING_METADATA
-            #  value: "false"
-            #- name: FORWARDING_OBJECT_THRESHOLD
-            #  value: "1"
-            #- name: REMOVE_AFTER_PROCESSED
-            #  value: "false"
-            #- name: NUMBER_OF_PREDICTIONS
-            #  valu
\ No newline at end of file
+            - name: FORWARDING_MEDIA
+              value: "True"
+            - name: REMOVE_AFTER_PROCESSED
+              value: "False"
\ No newline at end of file