12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- FROM tensorflow/tensorflow
- ENV WORKPATH /tensorflow
- WORKDIR $WORKPATH
- COPY . $WORKPATH/
- EXPOSE 8888 6006
- RUN apt-get update
- RUN apt-get upgrade -y
- RUN apt-get install -y git
- RUN git clone --depth 1 https://github.com/tensorflow/models.git
- RUN mv object_detection_wildfire.ipynb /tensorflow/models/research/object_detection/ && \
- mv create_tf_record.py /tensorflow/models/research/object_detection/
- RUN apt-get install -y protobuf-compiler python-pil python-lxml python-tk && \
- pip install Cython && \
- pip install contextlib2 && \
- pip install jupyter && \
- pip install matplotlib
- RUN git clone --depth 1 https://github.com/cocodataset/cocoapi.git && \
- cd cocoapi/PythonAPI && \
- make -j8 && \
- cp -r pycocotools /tensorflow/models/research && \
- cd ../../ && \
- rm -rf cocoapi
- RUN curl -OL "https://github.com/google/protobuf/releases/download/v3.0.0/protoc-3.0.0-linux-x86_64.zip" && \
- unzip protoc-3.0.0-linux-x86_64.zip -d proto3 && \
- mv proto3/bin/* /usr/local/bin && \
- mv proto3/include/* /usr/local/include && \
- rm -rf proto3 protoc-3.0.0-linux-x86_64.zip
- RUN cd /tensorflow/models/research && \
- protoc object_detection/protos/*.proto --python_out=.
- ENV PYTHONPATH $PYTHONPATH:/tensorflow/models/research:/tensorflow/models/research/slim
- RUN apt-get install -y wget vim emacs nano
- RUN curl -O "http://download.tensorflow.org/models/object_detection/faster_rcnn_resnet101_coco_11_06_2017.tar.gz" && \
- mkdir train && \
- tar xzf faster_rcnn_resnet101_coco_11_06_2017.tar.gz && \
- rm faster_rcnn_resnet101_coco_11_06_2017.tar.gz && \
- mv faster_rcnn_resnet101_coco_11_06_2017 train/ && \
- mv faster_rcnn_resnet101.config train/
- ENTRYPOINT ["/bin/bash"]
|