Install opencv for darknet models (#218)

* Install opencv for darknet models * Install ubuntu package for opencv
neo-ai · Jul 21, 2020 · c836984 · c836984
1 parent 650539d
commit c836984
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/container/Dockerfile.cpu b/container/Dockerfile.cpu
@@ -59,12 +59,13 @@ RUN apt-get update && \
     openjdk-8-jdk-headless \
     python3-dev \
     gcc \
+    libgtk2.0-dev \
     && rm -rf /var/lib/apt/lists/*
 
 COPY --from=builder /workspace/python/dist/*.whl /home/model-server/
 
 RUN pip3 install --pre --no-cache-dir multi-model-server \
-    && pip3 install --no-cache-dir --prefer-binary numpy scipy xlrd Pillow boto3 six requests mxnet tensorflow \
+    && pip3 install --no-cache-dir --prefer-binary numpy scipy xlrd Pillow boto3 six requests mxnet tensorflow opencv-python \
     && pip3 install /home/model-server/dlr-*.whl \
     && rm -rf /root/.cache/pip
 

diff --git a/container/Dockerfile.gpu b/container/Dockerfile.gpu
@@ -66,12 +66,13 @@ RUN apt-get update && \
     openjdk-8-jdk-headless \
     python3-dev \
     gcc \
+    libgtk2.0-dev \
     && rm -rf /var/lib/apt/lists/*
 
 COPY --from=builder /workspace/python/dist/*.whl /home/model-server/
 
 RUN pip3 install --pre --no-cache-dir multi-model-server \
-    && pip3 install --no-cache-dir --prefer-binary numpy scipy xlrd Pillow boto3 six requests mxnet-cu100 tensorflow_gpu \
+    && pip3 install --no-cache-dir --prefer-binary numpy scipy xlrd Pillow boto3 six requests mxnet-cu100 tensorflow_gpu opencv-python \
     && pip3 install /home/model-server/dlr-*.whl \
     && rm -rf /root/.cache/pip