DeepDetect deep learning server & API official GPU image
Requires nvidia-docker:
nvidia-docker: https://github.com/NVIDIA/nvidia-dockerNotes:
nvidia-docker requires docker >= 1.9Use as follows:
docker pull beniz/deepdetect_gpu
docker run -d -p 8080:8080 beniz/deepdetect_gpu
curl http://localhost:8080/info
{"status":{"code":200,"msg":"OK"},"head":{"method":"/info","version":"0.1","branch":"master","commit":"c8556f0b3e7d970bcd9861b910f9eae87cfd4b0c","services":[]}}
The image embeds googlenet and resnet_50 image classification models, see how to use them without effort below.
Here is how to do a simple image classification service and prediction test:
curl -X PUT "http://localhost:8080/services/imageserv" -d "{\"mllib\":\"caffe\",\"description\":\"image classification service\",\"type\":\"supervised\",\"parameters\":{\"input\":{\"connector\":\"image\"},\"mllib\":{\"nclasses\":1000}},\"model\":{\"repository\":\"/opt/models/ggnet/\"}}"
{"status":{"code":201,"msg":"Created"}}
curl -X POST "http://localhost:8080/predict" -d "{\"service\":\"imageserv\",\"parameters\":{\"input\":{\"width\":224,\"height\":224},\"output\":{\"best\":3},\"mllib\":{\"gpu\":true}},\"data\":[\"http://i.ytimg.com/vi/0vxOhd4qlnA/maxresdefault.jpg\"]}"
{"status":{"code":200,"msg":"OK"},"head":{"method":"/predict","time":852.0,"service":"imageserv"},"body":{"predictions":{"uri":"http://i.ytimg.com/vi/0vxOhd4qlnA/maxresdefault.jpg","classes":[{"prob":0.2255125343799591,"cat":"n03868863 oxygen mask"},{"prob":0.20917612314224244,"cat":"n03127747 crash helmet"},{"last":true,"prob":0.07399296760559082,"cat":"n03379051 football helmet"}]}}}
To look at server logs, use
docker logs -f <container name>
where can be obtained via docker ps
Content type
Image
Digest
Size
4.5 GB
Last updated
over 6 years ago
docker pull beniz/deepdetect_gpu