diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index a32f62418..65f57c997 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -50,7 +50,7 @@ jobs: uses: docker/build-push-action@v4 with: context: . - file: {context}/docker/build/main/Dockerfile + file: docker/build/main/Dockerfile push: true platforms: linux/amd64,linux/arm64 target: frigate @@ -62,7 +62,7 @@ jobs: uses: docker/build-push-action@v4 with: context: . - file: {context}/docker/build/rpi/Dockerfile + file: docker/build/rpi/Dockerfile push: true platforms: linux/arm64 target: frigate-rpi @@ -75,7 +75,7 @@ jobs: uses: docker/build-push-action@v4 with: context: . - file: {context}/docker/build/main/Dockerfile + file: docker/build/main/Dockerfile push: true platforms: linux/amd64 target: frigate-tensorrt diff --git a/docker-compose.yml b/docker-compose.yml index cf3613484..e79a8ab87 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -11,6 +11,7 @@ services: shm_size: "256mb" build: context: . + file: docker/build/main/Dockerfile # Use target devcontainer-trt for TensorRT dev target: devcontainer deploy: diff --git a/docker/build/main/Dockerfile b/docker/build/main/Dockerfile index 0ad4d9ca5..bb518909c 100644 --- a/docker/build/main/Dockerfile +++ b/docker/build/main/Dockerfile @@ -34,6 +34,32 @@ RUN wget -qO go2rtc "https://github.com/AlexxIT/go2rtc/releases/download/v1.6.2/ && chmod +x go2rtc +#### +# +# OpenVino Support +# +# 1. Download and convert a model from Intel's Public Open Model Zoo +# 2. Build libUSB without udev to handle NCS2 enumeration +# +#### +# Download and Convert OpenVino model +FROM base_amd64 AS ov-converter +ARG DEBIAN_FRONTEND + +# Install OpenVino Runtime and Dev library +COPY requirements-ov.txt /requirements-ov.txt +RUN apt-get -qq update \ + && apt-get -qq install -y wget python3 python3-distutils \ + && wget -q https://bootstrap.pypa.io/get-pip.py -O get-pip.py \ + && python3 get-pip.py "pip" \ + && pip install -r /requirements-ov.txt + +# Get OpenVino Model +RUN mkdir /models \ + && cd /models && omz_downloader --name ssdlite_mobilenet_v2 \ + && cd /models && omz_converter --name ssdlite_mobilenet_v2 --precision FP16 + + # libUSB - No Udev FROM wget as libusb-build ARG TARGETARCH