Skip to content

Commit dfd6913

Browse files
committed
Update Dockerfile for tf-faster-rcnn
Signed-off-by: Beat Buesser <[email protected]>
1 parent 27382ae commit dfd6913

File tree

1 file changed

+10
-22
lines changed

1 file changed

+10
-22
lines changed
Lines changed: 10 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,7 @@
1-
# Get base from a tensorflow image
2-
FROM tensorflow/tensorflow:1.15.5-py3
1+
# Get base from a TensorFlow image
2+
FROM python:3.7.17-slim
33

4-
# Set to install things in non-interactive mode
5-
ENV DEBIAN_FRONTEND noninteractive
6-
7-
# Install system wide softwares
4+
# Install system wide software
85
RUN apt-get update \
96
&& apt-get install -y \
107
libgl1-mesa-glx \
@@ -19,29 +16,20 @@ RUN apt-get update \
1916
libsndfile1 \
2017
wget \
2118
unzip \
19+
gpg \
20+
gpg-agent \
2221
&& apt-get clean all \
2322
&& rm -r /var/lib/apt/lists/*
2423

25-
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
26-
RUN bash Miniconda3-latest-Linux-x86_64.sh -b -p /miniconda
27-
RUN /miniconda/bin/conda install --yes \
28-
astropy \
29-
matplotlib \
30-
pandas \
31-
scikit-learn \
32-
scikit-image
33-
34-
# Install necessary libraries for tensorflow faster rcnn
24+
# Install necessary libraries for TensorFlow Faster-RCNN
3525
RUN mkdir TensorFlow
3626
RUN cd TensorFlow && git clone https://github.com/tensorflow/models
3727
RUN cd TensorFlow && wget https://github.com/protocolbuffers/protobuf/releases/download/v3.17.3/protoc-3.17.3-linux-x86_64.zip
3828
RUN cd TensorFlow && unzip protoc-3.17.3-linux-x86_64.zip -d protobuf
3929
RUN cd TensorFlow/models/research && /TensorFlow/protobuf/bin/protoc object_detection/protos/*.proto --python_out=.
4030
RUN cd TensorFlow/models/research && cp object_detection/packages/tf1/setup.py .
41-
RUN cd TensorFlow/models/research && python -m pip install --use-feature=2020-resolver .
31+
RUN cd TensorFlow/models/research && python -m pip install .
32+
33+
RUN pip install tensorflow==1.15.5 keras==2.2.5 tqdm==4.66.1 requests~=2.31.0 scikit-learn==1.0.2 pytest-cov~=4.1.0 protobuf==3.20
4234

43-
RUN pip install tqdm
44-
RUN pip install requests
45-
RUN pip install sklearn
46-
RUN pip install numba==0.50.0
47-
RUN pip install pytest-cov
35+
CMD ["/bin/bash"]

0 commit comments

Comments
 (0)