From 48e2c839f0b837fb38dd9e8a98d9321ba84ba079 Mon Sep 17 00:00:00 2001 From: Roman Donchenko Date: Fri, 4 Aug 2023 18:09:56 +0300 Subject: [PATCH] Remove the YOLOv5 dependency from the YOLOv7 serverless function For some reason, the YOLOv7 function uses `ultralytics/yolov5` as the base image, despite not depending on YOLOv5 in any way. Replace it with generic base images. --- serverless/onnx/WongKinYiu/yolov7/nuclio/function-gpu.yaml | 6 +++--- serverless/onnx/WongKinYiu/yolov7/nuclio/function.yaml | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/serverless/onnx/WongKinYiu/yolov7/nuclio/function-gpu.yaml b/serverless/onnx/WongKinYiu/yolov7/nuclio/function-gpu.yaml index 45f410aeb7ee..199c0d0c2f86 100644 --- a/serverless/onnx/WongKinYiu/yolov7/nuclio/function-gpu.yaml +++ b/serverless/onnx/WongKinYiu/yolov7/nuclio/function-gpu.yaml @@ -96,18 +96,18 @@ spec: eventTimeout: 30s build: image: cvat.onnx.wongkinyiu.yolov7 - baseImage: ultralytics/yolov5:latest + baseImage: nvidia/cuda:12.2.0-runtime-ubuntu22.04 directives: preCopy: - kind: USER value: root - kind: RUN - value: apt update && apt install --no-install-recommends -y libglib2.0-0 wget + value: apt update && apt install --no-install-recommends -y wget python3-pip - kind: WORKDIR value: /opt/nuclio - kind: RUN - value: pip install onnxruntime + value: pip install onnxruntime opencv-python-headless pillow pyyaml - kind: WORKDIR value: /opt/nuclio - kind: RUN diff --git a/serverless/onnx/WongKinYiu/yolov7/nuclio/function.yaml b/serverless/onnx/WongKinYiu/yolov7/nuclio/function.yaml index 328e2bbd64fc..5642ce58fb1b 100644 --- a/serverless/onnx/WongKinYiu/yolov7/nuclio/function.yaml +++ b/serverless/onnx/WongKinYiu/yolov7/nuclio/function.yaml @@ -95,16 +95,16 @@ spec: eventTimeout: 30s build: image: cvat.onnx.wongkinyiu.yolov7 - baseImage: ultralytics/yolov5:latest-cpu + baseImage: ubuntu:22.04 directives: preCopy: - kind: USER value: root - kind: RUN - value: apt update && apt install --no-install-recommends -y libglib2.0-0 && apt install wget + value: apt update && apt install --no-install-recommends -y wget python3-pip - kind: RUN - value: pip install onnxruntime + value: pip install onnxruntime opencv-python-headless pillow pyyaml - kind: WORKDIR value: /opt/nuclio - kind: RUN