diff --git a/base/hub/kerberos-hub-yolov8.yaml b/base/hub/kerberos-hub-yolov8.yaml index a4f61bc..ef04704 100644 --- a/base/hub/kerberos-hub-yolov8.yaml +++ b/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" \ No newline at end of file + - name: COLOR_PREDICTION_INTERVAL + value: "5" + - name: MIN_CLUSTERS + value: "4" + - name: MAX_CLUSTERS + value: "4" \ No newline at end of file diff --git a/base/hub/kustomization.yaml b/base/hub/kustomization.yaml index 4feb0a4..7d21a98 100644 --- a/base/hub/kustomization.yaml +++ b/base/hub/kustomization.yaml @@ -1,4 +1,5 @@ namespace: kerberos-hub resources: - kerberos-hub-nodeports.yaml - - kerberos-hub-import-database-job.yaml \ No newline at end of file + - kerberos-hub-import-database-job.yaml + - kerberos-hub-yolov8.yaml \ No newline at end of file