Project

deepstack

0.0
No release in over a year
A Ruby wrapper for DeepStack API
2005
2006
2007
2008
2009
2010
2011
2012
2013
2014
2015
2016
2017
2018
2019
2020
2021
2022
2023
2024
 Dependencies

Development

~> 13.0
~> 3.0
~> 1.21
>= 0
 Project Readme

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