Dockerize demo

tryolabs · Mar 1, 2024 · 20949c0 · 20949c0
1 parent 5c7b4c5
commit 20949c0
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 0 deletions.
diff --git a/demos/multi_camera/Dockerfile b/demos/multi_camera/Dockerfile
@@ -0,0 +1,6 @@
+FROM ultralytics/yolov5:v6.2
+
+# Install Norfair
+RUN pip install git+https://github.com/tryolabs/norfair.git@master#egg=norfair
+
+WORKDIR /demo/src/
diff --git a/demos/multi_camera/requirements.txt b/demos/multi_camera/requirements.txt
@@ -0,0 +1 @@
+yolov5==6.1.8
diff --git a/demos/multi_camera/run_gpu.sh b/demos/multi_camera/run_gpu.sh
@@ -0,0 +1,8 @@
+#!/usr/bin/env -S bash -e
+docker build . -t norfair-multicamera
+docker run -it --rm \
+           --gpus all \
+           --shm-size=1gb \
+           -v `realpath .`:/demo \
+           norfair-multicamera \
+           bash
diff --git a/demos/multi_camera/demo.py → demos/multi_camera/src/demo.py b/demos/multi_camera/demo.py → demos/multi_camera/src/demo.py