From 7cfc768cf71d63a4b2bc907dd47cbdc092273bbc Mon Sep 17 00:00:00 2001 From: thulyubh22 Date: Thu, 26 Oct 2023 15:58:29 +0800 Subject: [PATCH] modified dockerfile --- Dockerfile | 28 ++++++++++++++-------------- 1 file changed, 14 insertions(+), 14 deletions(-) diff --git a/Dockerfile b/Dockerfile index 8764e001..a88e0dff 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,38 +1,38 @@ -FROM nvidia/cuda:11.1-base-ubuntu20.04 +FROM nvidia/cuda:12.1.0-base-ubuntu20.04 RUN apt update && DEBIAN_FRONTEND=noninteractive apt install git bzip2 wget unzip python3-pip python3-dev cmake libgl1-mesa-dev python-is-python3 libgtk2.0-dev -yq -ADD . /app -WORKDIR /app -RUN cd Face_Enhancement/models/networks/ &&\ + +RUN git clone https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life.git + +RUN cd Bringing-Old-Photos-Back-to-Life/Face_Enhancement/models/networks/ &&\ git clone https://github.com/vacancy/Synchronized-BatchNorm-PyTorch &&\ cp -rf Synchronized-BatchNorm-PyTorch/sync_batchnorm . &&\ cd ../../../ -RUN cd Global/detection_models &&\ +RUN cd Bringing-Old-Photos-Back-to-Life/Global/detection_models &&\ git clone https://github.com/vacancy/Synchronized-BatchNorm-PyTorch &&\ cp -rf Synchronized-BatchNorm-PyTorch/sync_batchnorm . &&\ cd ../../ -RUN cd Face_Detection/ &&\ +RUN cd Bringing-Old-Photos-Back-to-Life/Face_Detection/ &&\ wget http://dlib.net/files/shape_predictor_68_face_landmarks.dat.bz2 &&\ bzip2 -d shape_predictor_68_face_landmarks.dat.bz2 &&\ cd ../ -RUN cd Face_Enhancement/ &&\ - wget https://facevc.blob.core.windows.net/zhanbo/old_photo/pretrain/Face_Enhancement/checkpoints.zip &&\ - unzip checkpoints.zip &&\ +RUN cd Bringing-Old-Photos-Back-to-Life/Face_Enhancement/ &&\ + wget https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life/releases/download/v1.0/face_checkpoints.zip &&\ + unzip face_checkpoints.zip &&\ cd ../ &&\ cd Global/ &&\ - wget https://facevc.blob.core.windows.net/zhanbo/old_photo/pretrain/Global/checkpoints.zip &&\ - unzip checkpoints.zip &&\ - rm -f checkpoints.zip &&\ + wget https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life/releases/download/v1.0/global_checkpoints.zip &&\ + unzip global_checkpoints.zip &&\ cd ../ RUN pip3 install numpy RUN pip3 install dlib -RUN pip3 install -r requirements.txt +RUN pip3 install -r Bringing-Old-Photos-Back-to-Life/requirements.txt RUN git clone https://github.com/NVlabs/SPADE.git @@ -40,4 +40,4 @@ RUN cd SPADE/ && pip3 install -r requirements.txt RUN cd .. -CMD ["python3", "run.py"] +CMD ["python3", "run.py"] \ No newline at end of file