tensorflow / models

Models and examples built with TensorFlow
Other
76.95k stars 45.79k forks source link

Dockerfile for DeepLabV3+ #9642

Open ssnirgudkar opened 3 years ago

ssnirgudkar commented 3 years ago

Prerequisites

Please answer the following question for yourself before submitting an issue.

1. The entire URL of the file you are using

https://github.com/tensorflow/models/tree/master/research/...

2. Describe the feature you request

I am trying to get DeepLabv3+ in a docker image on Ubuntu 18. However, I am hitting one issue after another. Refer to (https://github.com/tensorflow/models/issues/9625).

3. Additional context

This is a blocking issue for me.

4. Are you willing to contribute it? Yes

I already copy/pasted my version of Dockerfile which does not work in the above issue.

ssnirgudkar commented 3 years ago

I was able to get my dockerfile working with the simple unit test 'deeplab/model_test.py'. Here is my dockerfile ///////////////////////////////////////////////////////////////////////////////////////// FROM tensorflow/tensorflow:1.15.0-gpu

ARG deeplabpath=/deeplab WORKDIR $deeplabpath ENV MPLCONFIGDIR=/tmp/matplotlib/images

RUN set -eu pipefail && \ export DEBIAN_FRONTEND=noninteractive && \ apt-get update && \ apt-get -y upgrade && \ apt-get -y install wget && \ apt-get -y install python-pip &&\ pip install --upgrade pip && \ apt-get -y install python-pil python-numpy &&\ pip install --user jupyter &&\ pip install --user matplotlib &&\ pip install --user PrettyTable &&\ pip install --upgrade tf_slim &&\ apt-get clean

ENV PYTHONPATH $PYTHONPATH:$deeplabpath/models/research:$deeplabpath/models/research/slim ////////////////////////////////////////////////////////////////////////////////////////

However, I am not sure if it will work or not when I start training new data set.

saberkun commented 3 years ago

It would be great that if you can help with a PR to the documentation or adding a docker reference. @YknZhu

ssnirgudkar commented 3 years ago

Sure, please let me know what I need to do.