diff --git a/Dockerfile b/Dockerfile index 3b46f20..eaa4af1 100644 --- a/Dockerfile +++ b/Dockerfile @@ -18,6 +18,6 @@ RUN python3 -m pip install scikit-build RUN git clone --recursive https://github.com/skvark/opencv-python.git RUN python3 -m pip install --upgrade pip RUN cd opencv-python && python3 -m pip wheel . --verbose && find . -name "opencv_python*.whl" | xargs python3 -m pip install -RUN git clone https://github.com/ultralytics/yolov5.git +RUN git clone https://github.com/ultralytics/yolov5.git && cd yolov5 && git checkout 91547edec1a33121b3607e8c946e051514182b4f RUN cd yolov5 && python3 -m pip install -r requirements.txt WORKDIR /yolov5