diff --git a/preprocessors/clothes-detector/Dockerfile b/preprocessors/clothes-detector/Dockerfile index 5feecb543..9aecd3305 100644 --- a/preprocessors/clothes-detector/Dockerfile +++ b/preprocessors/clothes-detector/Dockerfile @@ -1,4 +1,4 @@ -FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-runtime +FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime RUN adduser --disabled-password python WORKDIR /app @@ -19,8 +19,8 @@ RUN apt-get install -y p7zip \ unace \ zip \ unzip -RUN pip install --upgrade pip -RUN pip3 install -r /app/requirements.txt +RUN pip3 install --upgrade pip +RUN pip install -r /app/requirements.txt RUN wget https://image.a11y.mcgill.ca/models/clothesDetector/yolo.zip RUN unzip yolo.zip diff --git a/preprocessors/clothes-detector/requirements.txt b/preprocessors/clothes-detector/requirements.txt index 32c814d20..04a0bf4e9 100644 --- a/preprocessors/clothes-detector/requirements.txt +++ b/preprocessors/clothes-detector/requirements.txt @@ -1,13 +1,13 @@ Flask==2.2.5 jsonschema==3.2.0 requests==2.31.0 -Werkzeug==2.3.8 +Werkzeug==2.2.3 gunicorn==20.1.0 -numpy==1.25 +numpy==1.21.0 opencv-python== 4.4.0.44 webcolors==1.12 -Pillow==10.2.0 +Pillow==9.4.0 torch==1.13.1 -matplotlib==3.8 +matplotlib==3.4.3 scikit-learn==1.0.1 colorthief \ No newline at end of file