Public Repository

Last pushed: 23 days ago
Short Description
Short description is empty for this repo.
Full Description

DeepDetect deep learning server & API official GPU image

Requires nvidia-docker:

Notes:

  • nvidia-docker requires docker >= 1.9

Use 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:

  • service creation
    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"}}
    
  • image classification
    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 <container name> can be obtained via docker ps

Docker Pull Command
Owner
beniz