Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

SSD: Single Shot MultiBox Detector for prediction with DD #213

Merged
merged 11 commits into from
Nov 30, 2016
Merged

SSD: Single Shot MultiBox Detector for prediction with DD #213

merged 11 commits into from
Nov 30, 2016

Conversation

beniz
Copy link
Collaborator

@beniz beniz commented Nov 21, 2016

This PR adds support for the image detection task (e.g. object detection) via SSD. Only prediction mode is covered by the present PR. Training should follow in a near futur.

Functionalities:

  • multi-class object detection
  • single and batch forward processing of images
  • JSON output with bouding boxes resized to fit the originally submitted image
  • threshold on prediction confidence via confidence_threshold output parameter
  • ready-to use demo in demo/objdetect

Sample usage:

  • create service

This is identical to other tasks

curl -X PUT "http://localhost:8080/services/imageserv" -d "{\"mllib\":\"caffe\",\"description\":\"image classification service\",\"type\":\"supervised\",\"parameters\":{\"input\":{\"connector\":\"image\",\"height\":300,\"width\":300},\"mllib\":{\"nclasses\":2}},\"model\":{\"repository\":\"/path/to/model\"}}"
  • prediction
curl -X POST "http://localhost:8080/predict" -d '{"service":"imageserv","parameters":{"output":{"bbox":true,"confidence_threshold":0.4}},"data":["/path/to/image"]}'

@beniz beniz merged commit d2673da into master Nov 30, 2016
@beniz beniz deleted the ssd branch February 5, 2017 16:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant