From f028c0f1eab765242d729221e269fd00de22b904 Mon Sep 17 00:00:00 2001 From: Niaz Faridani-Rad <faridani.rad@gmail.com> Date: Wed, 9 Jun 2021 01:53:42 +0200 Subject: [PATCH] [FIX] Change models download path to our repository - also changed to slim buster to decrease the download size --- backend/Dockerfile | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/backend/Dockerfile b/backend/Dockerfile index 0fb3587..c82b821 100644 --- a/backend/Dockerfile +++ b/backend/Dockerfile @@ -1,17 +1,14 @@ -FROM python:3.8 +FROM python:3.8-slim-buster # system packages installation -RUN apt update && apt install -y ffmpeg libimage-exiftool-perl curl nfs-common cifs-utils libopenblas-dev libmagickwand-dev libheif-dev libmagic1 ufraw-batch libboost-all-dev libxrender-dev liblapack-dev git bzip2 cmake build-essential libsm6 libglib2.0-0 libgl1-mesa-glx --no-install-recommends +RUN apt update && apt install -y ffmpeg libimage-exiftool-perl curl nfs-common cifs-utils libopenblas-dev libmagickwand-dev libheif-dev libmagic1 ufraw-batch libboost-all-dev libxrender-dev liblapack-dev git bzip2 cmake build-essential libsm6 libglib2.0-0 libgl1-mesa-glx libpq-dev --no-install-recommends # pre trained models download WORKDIR /data_models -RUN mkdir -p /data_models/places365/ -RUN mkdir -p /data_models/im2txt/ RUN mkdir -p /root/.cache/torch/hub/checkpoints/ -RUN curl -SL https://s3.eu-central-1.amazonaws.com/ownphotos-deploy/places365_model.tar.gz | tar -zxC /data_models/places365/ -RUN curl -SL https://s3.eu-central-1.amazonaws.com/ownphotos-deploy/im2txt_model.tar.gz | tar -zxC /data_models/im2txt/ -RUN curl -SL https://s3.eu-central-1.amazonaws.com/ownphotos-deploy/im2txt_data.tar.gz | tar -zxC /data_models/im2txt/ +RUN curl -SL https://github.com/LibrePhotos/librephotos-docker/releases/download/0.1/places365.tar.gz | tar -zxC /data_models/ +RUN curl -SL https://github.com/LibrePhotos/librephotos-docker/releases/download/0.1/im2txt.tar.gz | tar -zxC /data_models/ RUN curl -SL https://download.pytorch.org/models/resnet152-b121ed2d.pth -o /root/.cache/torch/hub/checkpoints/resnet152-b121ed2d.pth RUN pip install torch==1.8.0+cpu torchvision==0.9.0+cpu -f https://download.pytorch.org/whl/torch_stable.html -- GitLab