search

microsoft / Bringing-Old-Photos-Back-to-Life

Bringing Old Photo Back to Life (CVPR 2020 oral)
https://arxiv.org/abs/2004.09484
MIT License
15.12k stars 1.99k forks source link

refined dockerfile #284

Open Imbernoulli opened 1 year ago

Imbernoulli commented 1 year ago

There are many problems with the old docker file, here is the new available one

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/ &&\
  git clone https://github.com/vacancy/Synchronized-BatchNorm-PyTorch &&\
  cp -rf Synchronized-BatchNorm-PyTorch/sync_batchnorm . &&\
  cd ../../../

RUN cd Global/detection_models &&\
  git clone https://github.com/vacancy/Synchronized-BatchNorm-PyTorch &&\
  cp -rf Synchronized-BatchNorm-PyTorch/sync_batchnorm . &&\
  cd ../../

RUN cd 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://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://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 git clone https://github.com/NVlabs/SPADE.git

RUN cd SPADE/ && pip3 install -r requirements.txt

RUN cd ..

CMD ["python3", "run.py"]
Samuel-Hinchliffe commented 1 year ago

You saved my day. 😁

Imbernoulli commented 1 year ago

You can also use this, which is not dependent (not using ADD) to this repo

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

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 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 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 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://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 Bringing-Old-Photos-Back-to-Life/requirements.txt

RUN git clone https://github.com/NVlabs/SPADE.git

RUN cd SPADE/ && pip3 install -r requirements.txt

RUN cd ..

CMD ["python3", "run.py"]