Merge pull request #8 from SoftArmpit/add-dockerfile

build: add dockerfile
This commit is contained in:
deeppomf 2018-06-19 15:46:57 -04:00 committed by GitHub
commit 56ca434ba6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 20 additions and 0 deletions

3
.gitignore vendored Normal file
View File

@ -0,0 +1,3 @@
models/
decensor_input/
decensor_output/

12
Dockerfile-py3 Normal file
View File

@ -0,0 +1,12 @@
FROM tensorflow/tensorflow:latest-py3
RUN apt update && apt install -y python3 python3-pip python3-tk python3-numpy && apt clean
WORKDIR /app
COPY . /app
RUN pip --no-cache-dir install --upgrade pip setuptools
RUN pip --no-cache-dir install wheel
RUN pip --no-cache-dir install -r requirements.txt
CMD ["python3", "decensor.py"]

5
requirements.txt Normal file
View File

@ -0,0 +1,5 @@
Pillow
tqdm
scipy
pyamg
matplotlib