opencv requirements and docker

1 year ago · d8f551a2f8
parent f5dfa0926f
commit d8f551a2f8
2 changed files with 3 additions and 1 deletions
--- a/2
+++ b/2
@ -2,6 +2,8 @@
 # ==================================
 # Use an official Python runtime as a parent image
 FROM python:3.9-slim
+RUN apt-get update && apt-get -y install libglib2.0-0; apt-get clean
+RUN pip install opencv-contrib-python-headless

 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE 1
--- a/requirements.txt
+++ b/requirements.txt
@ -38,7 +38,7 @@ controlnet-aux
 diffusers
 einops
 imageio
-opencv-python-headless
+opencv-python-headless=="*"
 imageio-ffmpeg
 invisible-watermark
 kornia