Dockerfile

FROM tensorflow/tensorflow:1.8.0-py3
LABEL maintainer="Rub21"
ENV workdir /usr/src/app
ENV TOD /usr/src/app/models/research/object_detection
RUN mkdir $workdir
RUN apt-get update -y
RUN apt-get install -y \
    libsqlite3-dev \
    libcurl4-openssl-dev \
    libssl-dev \
    wget \
    git

# Install tippecanoe
RUN git clone --progress https://github.com/mapbox/tippecanoe.git && \
    cd tippecanoe && \
    make -j && \
    make install

# Install label-maker
RUN pip install \
    label-maker==0.3.1 --ignore-installed pycurl

# Install remaining python dependencies
RUN pip install \
    Cython==0.28.2 \
    lxml==4.2.1

# Install Proto
RUN wget https://github.com/google/protobuf/releases/download/v3.2.0/protoc-3.2.0-linux-x86_64.zip -P $workdir
RUN unzip $workdir/protoc-3.2.0-linux-x86_64.zip -d $workdir/protoc3
RUN mv -f $workdir/protoc3/bin/* /usr/local/bin/
RUN mv -f $workdir/protoc3/include/* /usr/local/include/
RUN ln -s -f /usr/local/bin/protoc /usr/bin/protoc

# Setup TensorFlow Object Detection API
RUN git clone --progress https://github.com/tensorflow/models.git $workdir/models && \
    cd $workdir/models && \
    git checkout ee968baddd96d9e909fdea3b2923a5a5ccebe724
WORKDIR $workdir
RUN cd $workdir/models/research && \
    protoc object_detection/protos/*.proto --python_out=.
ENV PYTHONPATH=$PYTHONPATH:/usr/src/app/models/research:/usr/src/app/models/research/slim
RUN cd $workdir/models/research && \
    python object_detection/builders/model_builder_test.py

# Object detection model setup
# https://github.com/developmentseed/label-maker/blob/master/examples/walkthrough-tensorflow-object-detection.md#object-detection-model-setup
RUN cd $WORKDIR && wget http://download.tensorflow.org/models/object_detection/ssd_inception_v2_coco_2017_11_17.tar.gz && \
    tar xvf ssd_inception_v2_coco_2017_11_17.tar.gz && \
    mv ssd_inception_v2_coco_2017_11_17 $TOD/

# Create TFRecords for model training
RUN git clone --progress https://github.com/developmentseed/label-maker.git $workdir/label-maker && \
    cd $workdir/label-maker && \
    git checkout 94f1863945c47e1b69fe0d6d575caa0b42aa8d63
COPY ./script.sh $workdir
CMD ["bash", "script.sh"]