Add docker configuration files for CPU and GPU.

docker-cpu.df

@@ -0,0 +1,32 @@
+FROM ubuntu:14.04
+
+# Install dependencies
+RUN apt-get -qq update           &&  \
+    apt-get -qq install --assume-yes \
+        "build-essential"            \
+        "git"                        \
+        "wget"                       \
+        "pkg-config"              && \
+    rm -rf /var/lib/apt/lists/*
+
+# Miniconda.
+RUN wget --quiet https://repo.continuum.io/miniconda/Miniconda2-4.2.11-Linux-x86_64.sh -O ~/miniconda.sh && \
+    /bin/bash ~/miniconda.sh -b -p /opt/conda && \
+    rm ~/miniconda.sh
+
+# Install requirements before copying project files
+WORKDIR /ne
+COPY requirements.txt .
+RUN /opt/conda/bin/conda install -q -y conda numpy scipy pip pillow
+RUN /opt/conda/bin/python3 -m pip install -q -r "requirements.txt"
+
+# Copy only required project files
+COPY enhance.py .
+
+# Get a pre-trained neural networks, non-commercial & attribution.
+RUN wget -q "https://github.com/alexjc/neural-enhance/releases/download/v0.1/ne4x-small-0.1.pkl.bz2"
+RUN wget -q "https://github.com/alexjc/neural-enhance/releases/download/v0.1/ne4x-medium-0.1.pkl.bz2"
+RUN wget -q "https://github.com/alexjc/neural-enhance/releases/download/v0.1/ne4x-large-0.1.pkl.bz2"
+
+# Set an entrypoint to the main doodle.py script
+ENTRYPOINT ["/opt/conda/bin/python3", "enhance.py", "--device=cpu"]

docker-gpu.df

@@ -0,0 +1,32 @@
+FROM nvidia/cuda:8.0-cudnn5-devel
+
+# Install dependencies
+RUN apt-get -qq update           &&  \
+    apt-get -qq install --assume-yes \
+        "build-essential"            \
+        "git"                        \
+        "wget"                       \
+        "pkg-config"              && \
+    rm -rf /var/lib/apt/lists/*
+
+# Miniconda.
+RUN wget --quiet https://repo.continuum.io/miniconda/Miniconda2-4.2.11-Linux-x86_64.sh -O ~/miniconda.sh && \
+    /bin/bash ~/miniconda.sh -b -p /opt/conda && \
+    rm ~/miniconda.sh
+
+# Install requirements before copying project files
+WORKDIR /ne
+COPY requirements.txt .
+RUN /opt/conda/bin/conda install -q -y conda numpy scipy pip pillow
+RUN /opt/conda/bin/python3 -m pip install -q -r "requirements.txt"
+
+# Copy only required project files
+COPY enhance.py .
+
+# Get a pre-trained neural networks, non-commercial & attribution.
+RUN wget -q "https://github.com/alexjc/neural-enhance/releases/download/v0.1/ne4x-small-0.1.pkl.bz2"
+RUN wget -q "https://github.com/alexjc/neural-enhance/releases/download/v0.1/ne4x-medium-0.1.pkl.bz2"
+RUN wget -q "https://github.com/alexjc/neural-enhance/releases/download/v0.1/ne4x-large-0.1.pkl.bz2"
+
+# Set an entrypoint to the main doodle.py script
+ENTRYPOINT ["/opt/conda/bin/python3", "enhance.py", "--device=gpu"]