ultralytics 8.0.96 TAL speed and memory improvements (#2484)

Signed-off-by: Evangelos Petrongonas <e.petrongonas@hellenicdrones.com>
Co-authored-by: Evangelos Petrongonas <24351757+vpetrog@users.noreply.github.com>
Co-authored-by: JF Chen <k-2feng@hotmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocher
2023-05-08 23:41:27 +02:00
committed by GitHub
parent e21428ca4e
commit 6ee3a9a74b
13 changed files with 163 additions and 53 deletions

View File

@ -28,6 +28,12 @@ RUN python3 -m pip install --upgrade pip wheel
RUN pip install --no-cache tqdm matplotlib pyyaml psutil thop pandas onnx "numpy==1.23"
RUN pip install --no-cache -e .
# Resolve duplicate OpenCV installation issues in https://github.com/ultralytics/ultralytics/issues/2407
RUN apt-get remove `dpkg -l | grep opencv | awk '{print $2}'`
RUN pip uninstall -y opencv-python
RUN rm /usr/local/lib/python3.8/dist-packages/cv2 # Optional
RUN pip install "opencv-python<4.7"
# Set environment variables
ENV OMP_NUM_THREADS=1