Skip to content

Commit

Permalink
Remove the YOLOv5 dependency from the YOLOv7 serverless function (#6619)
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
SpecLad authored Aug 7, 2023
1 parent b5223b6 commit a8e921b
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
6 changes: 3 additions & 3 deletions serverless/onnx/WongKinYiu/yolov7/nuclio/function-gpu.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
6 changes: 3 additions & 3 deletions serverless/onnx/WongKinYiu/yolov7/nuclio/function.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit a8e921b

Please sign in to comment.