DeepStack Ruby Wrapper

A Ruby wrapper for DeepStack HTTP API

Usage

Install the gem with gem install deepstack

require 'deepstack'

deepstack = DeepStack.new('http://192.168.1.10:2000')
image = File.read('image.jpg')

# Find bounding rects for objects
predictions = deepstack.detect_objects(image) 
# => [{"confidence"=>0.86599416, "label"=>"dog", "y_min"=>355, "x_min"=>648, "y_max"=>540, "x_max"=>797},
#     {"confidence"=>0.918332, "label"=>"person", "y_min"=>113, "x_min"=>442, "y_max"=>524, "x_max"=>601},
#     {"confidence"=>0.9292374, "label"=>"person", "y_min"=>83, "x_min"=>294, "y_max"=>521, "x_max"=>447}]

# Find bounding rects for faces
faces = deepstack.detect_faces(image)
# => [{"confidence"=>0.86419886, "y_min"=>236, "x_min"=>876, "y_max"=>730, "x_max"=>1203},
#     {"confidence"=>0.8811783, "y_min"=>164, "x_min"=>1617, "y_max"=>692, "x_max"=>1985}]

# Register a face for face recognition
deepstack.register_face('face1', image)

# List all the registered faces in the system
face_list = deepstack.face_list
# => ["face1", "face2"]

# Perform a face recognition, return identified userids
faces = deepstack.recognize_face(image)
# => [{"confidence"=>0, "userid"=>"unknown", "y_min"=>236, "x_min"=>876, "y_max"=>730, "x_max"=>1203},
#     {"confidence"=>0.9824197, "userid"=>"face1", "y_min"=>164, "x_min"=>1617, "y_max"=>692, "x_max"=>1985}]

# Delete a face from the system
deepstack.delete_face('face1')

# Detect similarities between two face images
image2 = File.read('image2.jpg')
similarity = deepstack.face_match(image, image2)
# => 0.3333

# Perform Scene recognition
scene = deepstack.identify_scene(image)
# => {"success"=>true, "confidence"=>0.27867314, "label"=>"archive", "duration"=>0}

Error Handling

The methods will:

  • Raise an exception on a connection error.
  • Return nil when DeepStack reports failure (success=false)

See the documentation for more details.

Development

A Linux development machine is needed in order to run the tests. The test will launch DeepStack docker instances to test against. By default, the deepstack docker will listen on ports 8001-8004. To change this, copy rakelib/deepstack.yml.sample to rakelib/deepstack.yml and change the port numbers as required.

To run the tests run rake.

To manually stop the DeepStack docker instance, run rake deepstack:stop

To install this gem onto your local machine, run bundle exec rake install.

Contributing

Bug reports and pull requests are welcome on GitHub.

License

The gem is available as open source under the terms of the EPL 2.0 License