Update data-filtering-deployment.yaml

This commit is contained in:
Cedric Verstraeten
2024-08-10 20:04:20 +02:00
parent c45ef0abda
commit aa02253ebc

View File

@@ -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
- name: FORWARDING_MEDIA
value: "True"
- name: REMOVE_AFTER_PROCESSED
value: "False"