Skip to content

Commit 0a349ee

Browse files
committed
changed Dockerfiles to clone r1.13.0 branch instead of master
1 parent 1362952 commit 0a349ee

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

docker/cpu/Dockerfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,14 +7,15 @@ RUN apt-get update && \
77
RUN apt-get install -y git
88

99
# Installing the Tensorflow Object Detection API (https://gilberttanner.com/blog/installing-the-tensorflow-object-detection-api)
10-
RUN git clone --depth 1 https://github.com/tensorflow/models.git
10+
RUN git clone --depth 1 -b r1.13.0 https://github.com/tensorflow/models.git
1111

1212
# Install object detection api dependencies
1313
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y protobuf-compiler python3-pil python3-lxml python3-tk && \
1414
pip install Cython && \
1515
pip install contextlib2 && \
1616
pip install jupyter && \
1717
pip install matplotlib && \
18+
pip install Pillow && \
1819
pip install tensorflow==1.14.0
1920

2021
# Install pycocoapi

docker/gpu/Dockerfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,14 +9,15 @@ RUN apt-get install -y python3 && apt-get install -y python3-pip
99
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.6 1
1010

1111
# Installing the Tensorflow Object Detection API (https://gilberttanner.com/blog/installing-the-tensorflow-object-detection-api)
12-
RUN git clone --depth 1 https://github.com/tensorflow/models.git
12+
RUN git clone --depth 1 -b r1.13.0 https://github.com/tensorflow/models.git
1313

1414
# Install object detection api dependencies
1515
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y protobuf-compiler python3-pil python3-lxml python3-tk && \
1616
pip3 install Cython && \
1717
pip3 install contextlib2 && \
1818
pip3 install jupyter && \
1919
pip3 install matplotlib && \
20+
pip3 install Pillow && \
2021
pip3 install tensorflow==1.14.0
2122

2223
# Install pycocoapi

0 commit comments

Comments
 (0)