Adding models to the docker container

[visionai]

Adding models to the docker container

[quantombone]

I would recommend that you take the visionai/clouddream Docker image and extend it by writing your own Dockerfile. From my understanding of how Docker Hub works, you should not be able to push a container under the visionai/clouddream because visionai is already reserved. For your own debugging sanity, I would recommend naming your custom image something different.

Let's first make a new directory for the new Dockerfile:

cd ~/clouddream/
mkdir custom
cd custom

Now's lets create a simple Dockerfile which extends the visionai one, but downloads the AlexNet model. The file looks like this:

FROM visionai/clouddream

#Download my custom model
RUN /opt/caffe/scripts/download_model_binary.py /opt/caffe/models/bvlc_alexnet

Now's let's build the thing and give it a name like custom/clouddream and we'll be using it on one machine so we won't distribute it to Docker Hub.

#Make sure you are in the custom directory, which has the short Dockerfile inside:
docker build -t custom/clouddream  .

You have now built a custom/clouddream container on your machine, and can go ahead and modify the references inside the Python code to your new model (AlexNet) instead of the default GoogLeNet one.