Skip to content

Commit

Permalink
Dockerfile and kubernetes example
Browse files Browse the repository at this point in the history
  • Loading branch information
hijak committed Nov 20, 2020
1 parent ab40e6b commit 77bce10
Showing 1 changed file with 43 additions and 0 deletions.
43 changes: 43 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,43 @@
FROM nvidia/cuda:11.1-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://facevc.blob.core.windows.net/zhanbo/old_photo/pretrain/Face_Enhancement/checkpoints.zip &&\
unzip 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 &&\
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"]

0 comments on commit 77bce10

Please sign in to comment.