diff --git a/Dockerfile b/Dockerfile index 209b729e..97ede3d4 100644 --- a/Dockerfile +++ b/Dockerfile @@ -9,28 +9,28 @@ FROM nvcr.io/nvidia/pytorch:19.08-py3 # conda install pytorch torchvision -c pytorch # Install OpenCV with Gstreamer support -#WORKDIR /usr/src -#RUN pip uninstall -y opencv-python -#RUN apt-get update -#RUN apt-get install -y gstreamer1.0-python3-dbg-plugin-loader -## RUN apt-get install gstreamer1.0 -## RUN apt install -y ubuntu-restricted-extras -#RUN apt install -y libgstreamer1.0-dev -#RUN apt install -y libgstreamer-plugins-base1.0-dev -#RUN git clone https://github.com/opencv/opencv.git && cd opencv && git checkout 4.1.1 && mkdir build -#RUN git clone https://github.com/opencv/opencv_contrib.git && cd opencv_contrib && git checkout 4.1.1 -#RUN cd opencv/build && cmake ../ \ -# -D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules \ -# -D BUILD_OPENCV_PYTHON3=ON \ -# -D PYTHON3_EXECUTABLE=/opt/conda/bin/python \ -# -D PYTHON3_INCLUDE_PATH=/opt/conda/include/python3.6m \ -# -D PYTHON3_LIBRARIES=/opt/conda/lib/python3.6/site-packages \ -# -D WITH_GSTREAMER=ON \ -# -D WITH_FFMPEG=OFF \ -# && make && make install && ldconfig -#RUN cd /usr/local/lib/python3.6/site-packages/ && mv cv2.cpython-36m-x86_64-linux-gnu.so cv2.so -#RUN cd /opt/conda/lib/python3.6/site-packages/ && ln -s /usr/local/lib/python3.6/site-packages/cv2.so cv2.so -#RUN python3 -c "import cv2; print(cv2.getBuildInformation())" +WORKDIR /usr/src +RUN pip uninstall -y opencv-python +RUN apt-get update +RUN apt-get install -y gstreamer1.0-python3-dbg-plugin-loader +# RUN apt-get install gstreamer1.0 +# RUN apt install -y ubuntu-restricted-extras +RUN apt install -y libgstreamer1.0-dev +RUN apt install -y libgstreamer-plugins-base1.0-dev +RUN git clone https://github.com/opencv/opencv.git && cd opencv && git checkout 4.1.1 && mkdir build +RUN git clone https://github.com/opencv/opencv_contrib.git && cd opencv_contrib && git checkout 4.1.1 +RUN cd opencv/build && cmake ../ \ + -D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules \ + -D BUILD_OPENCV_PYTHON3=ON \ + -D PYTHON3_EXECUTABLE=/opt/conda/bin/python \ + -D PYTHON3_INCLUDE_PATH=/opt/conda/include/python3.6m \ + -D PYTHON3_LIBRARIES=/opt/conda/lib/python3.6/site-packages \ + -D WITH_GSTREAMER=ON \ + -D WITH_FFMPEG=OFF \ + && make && make install && ldconfig +RUN cd /usr/local/lib/python3.6/site-packages/cv2/python-3.6/ && mv cv2.cpython-36m-x86_64-linux-gnu.so cv2.so +RUN cd /opt/conda/lib/python3.6/site-packages/ && ln -s /usr/local/lib/python3.6/site-packages/cv2/python-3.6/cv2.so cv2.so +RUN python3 -c "import cv2; print(cv2.getBuildInformation())" # Create working directory RUN mkdir -p /usr/src/app