Commit 54c5288b authored by Ketil Malde's avatar Ketil Malde
Browse files

Download stuff to /src and make things work

parent de13e372
......@@ -42,7 +42,7 @@ default: .docker
# $(RUNCMD) python3 src/download_weights.py
train: .docker $(WEIGHTS)
$(RUNCMD) python3 keras-retinanet/keras_retinanet/bin/train.py
$(RUNCMD) python3 /src/keras-retinanet/keras_retinanet/bin/train.py csv /data/train.csv /data/classes.csv --val-annotations /data/validation/validation.csv
test: .docker src/test.py $(WEIGHTS)
$(RUNCMD) python3 src/test.py
......@@ -23,10 +23,10 @@ RUN useradd -u $uid -g $gid $user
RUN usermod -a -G sudo $user
RUN passwd -d $user
# Alternative here, but let's run in place (below)
# RUN pip3 install . --user
RUN git clone https://git.imr.no/a5515/keras-retinanet.git
RUN cd keras-retinanet && python setup.py build_ext --inplace
# Need to do both installs to make it work
RUN mkdir /src && cd /src && git clone https://git.imr.no/a5515/keras-retinanet.git
RUN cd /src/keras-retinanet && python setup.py build_ext --inplace
RUN cd /src/keras-retinanet && pip3 install .
RUN apt-get install -y libsm6 libxrender1
# keras 2.4.0 requires tensorflow 2.2, but Matterport Mask R-CNN doesn't support it
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment