Dockerfile for running the cv-tricks image classifier tensorflow tutorial

Dockerfile

# Dockerfile for running the cv-tricks image classifier tensorflow tutorial from
# http://cv-tricks.com/tensorflow-tutorial/training-convolutional-neural-network-for-image-classification/

FROM tensorflow/tensorflow:latest-gpu-py3
# or just tensorflow/tensorflow:latest-py3 for CPU version

LABEL maintainer="Sebastian Stammler <base64 -d<<<aWFtQGRlY2VudHIuYWwK>"

RUN apt-get update && \
  apt-get install -y --no-install-recommends \
  libsm6 \
  libxrender1 \
  libxext6 \
  && rm -rf /var/lib/apt/lists/*

RUN pip --no-cache-dir install opencv-python

COPY cv-tricks.com/Tensorflow-tutorials/tutorial-2-image-classifier /app

WORKDIR /app

CMD python train.py