deepdetect_ruby 1.0.1

data/docs/finetuning_googlenet.py

@@ -0,0 +1,40 @@
+from dd_client import DD
+
+model_repo = '/home/me/models/mymodel'
+height = width = 224
+nclasses = 300
+
+# setting up DD client
+host = '127.0.0.1'
+sname = 'imgserv'
+description = 'image classification'
+mllib = 'caffe'
+dd = DD(host)
+dd.set_return_format(dd.RETURN_PYTHON)
+
+# creating ML service
+model = {'repository':model_repo}
+parameters_input = {'connector':'image','width':width,'height':height}
+parameters_mllib = {'template':'googlenet','nclasses':nclasses,'finetuning':True,'rotate':False,'mirror':True,'weights':'bvlc_googlenet.caffemodel'}
+parameters_output = {}
+dd.put_service(sname,model,description,mllib,
+               parameters_input,parameters_mllib,parameters_output)
+
+# training / finetuning from pre-trained network
+train_data = ['path/to/train/data']
+parameters_input = {'test_split':0.1,'shuffle':True}
+parameters_mllib = {'gpu':True,'net':{'batch_size':64,'test_batch_size':64},'solver':{'test_interval':4000,'iterations':50000,'snapshot':10000,'base_lr':0.01,'solver_type':'NESTEROV','test_initialization':True}}
+parameters_output = {'measure':['mcll','f1','acc-5']}
+dd.post_train(sname,train_data,parameters_input,parameters_mllib,parameters_output,async=True)
+
+time.sleep(1)
+train_status = ''
+while True:
+    train_status = dd.get_train(sname,job=1,timeout=10)
+    if train_status['head']['status'] == 'running':
+  print train_status['body']['measure']
+    else:
+  print train_status
+  break
+
+dd.delete_service(sname)

data/docs/rsync_for_selfie_f4.sh

@@ -0,0 +1,23 @@
+# rsync for selfie f40
+echo "---> sync selfie #0"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect_u1/models/selfie_f40/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie/
+
+echo "---> sync selfie #1"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect_u1/models/selfie_f40/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_1/
+
+echo "---> sync selfie #2"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect_u1/models/selfie_40/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_2/
+
+echo "---> sync selfie #3"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect_u1/models/selfie_f40/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_3/
+
+echo "---> sync selfie #4"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect_u1/models/selfie_f40/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_4/
+
+echo "---> sync selfie #5"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect_u1/models/selfie_f40/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_5/
+
+echo "---> sync selfie #reanalyze"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect_u1/models/selfie_f40/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_reanalyze/
+
+echo -e "\nsync fine-tuned selfie f40 done! \n"

data/docs/sync_file.sh

@@ -0,0 +1,100 @@
+# rsync -avh /home/ubuntu/downloads/test ubuntu@54.229.220.160:/home/ubuntu/downloads/test
+
+# rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/person_yes_no_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/person_yes_no_f1/
+
+echo "---> sync person_yes_no"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/person_yes_no_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/person_yes_no/
+
+echo "---> sync person_yes_no #1"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/person_yes_no_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/person_yes_no_1/
+
+echo "---> sync person_yes_no #2"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/person_yes_no_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/person_yes_no_2/
+
+echo "---> sync person_yes_no #3"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/person_yes_no_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/person_yes_no_3/
+
+echo "---> sync person_yes_no #4"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/person_yes_no_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/person_yes_no_4/
+
+echo "---> sync person_yes_no #5"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/person_yes_no_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/person_yes_no_5/
+
+echo "---> sync person_yes_no #reanalyze"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/person_yes_no_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/person_yes_no_reanalyze/
+
+# rsync for selfie f1
+echo "---> sync selfie"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/selfie_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie/
+
+echo "---> sync selfie #1"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/selfie_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_1/
+
+echo "---> sync selfie #2"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/selfie_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_2/
+
+echo "---> sync selfie #3"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/selfie_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_3/
+
+echo "---> sync selfie #4"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/selfie_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_4/
+
+echo "---> sync selfie #5"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/selfie_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_5/
+
+echo "---> sync selfie #reanalyze"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/selfie_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_reanalyze/
+
+echo -e "\nsync fine-tuned selfie done! \n"
+
+# rsync for selfie f2
+echo "---> sync selfie f2"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/selfie_f2/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie/
+
+echo "---> sync selfie #1"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/selfie_f2/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_1/
+
+echo "---> sync selfie #2"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/selfie_f2/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_2/
+
+echo "---> sync selfie #3"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/selfie_f2/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_3/
+
+echo "---> sync selfie #4"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/selfie_f2/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_4/
+
+echo "---> sync selfie #5"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/selfie_f2/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_5/
+
+echo "---> sync selfie #reanalyze"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/selfie_f2/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/selfie_reanalyze/
+
+echo -e "\nsync fine-tuned selfie f2 done! \n"
+
+# --------------------------------------------------
+# rsync for more_than_one_f1
+echo "---> sync more_than_one_f1"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/more_than_one_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/more_than_one/
+
+echo "---> sync more_than_one #1"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/more_than_one_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/more_than_one_1/
+
+echo "---> sync more_than_one #2"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/more_than_one_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/more_than_one_2/
+
+echo "---> sync more_than_one #3"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/more_than_one_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/more_than_one_3/
+
+echo "---> sync more_than_one #4"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/more_than_one_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/more_than_one_4/
+
+echo "---> sync more_than_one #5"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/more_than_one_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/more_than_one_5/
+
+echo "---> sync more_than_one #reanalyze"
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/more_than_one_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/more_than_one_reanalyze/
+
+echo -e "\nsync fine-tuned more_than_one_f1 done! \n"
+
+
+rsync -arvu --delete /home/ubuntu/projects/deepdetect/models/person_yes_no_f1/ ubuntu@54.229.220.160:/home/ubuntu/projects/deepdetect/models/person_yes_no_1/

data/docs/sync_mirrorshot.sh

@@ -0,0 +1,23 @@
+# rsync for mirrorshot f1
+echo "---> sync mirrorshot #0"
+rsync -arvu --delete /data/projects/deepdetect_u1/models/mirrorshot_f1/ ubuntu@52.51.246.115:/home/ubuntu/projects/deepdetect/models/mirrorshot/
+
+echo "---> sync mirrorshot #1"
+rsync -arvu --delete /data/projects/deepdetect_u1/models/mirrorshot_f1/ ubuntu@52.51.246.115:/home/ubuntu/projects/deepdetect/models/mirrorshot_1/
+
+echo "---> sync mirrorshot #2"
+rsync -arvu --delete /data/projects/deepdetect_u1/models/mirrorshot_f1/ ubuntu@52.51.246.115:/home/ubuntu/projects/deepdetect/models/mirrorshot_2/
+
+echo "---> sync mirrorshot #3"
+rsync -arvu --delete /data/projects/deepdetect_u1/models/mirrorshot_f1/ ubuntu@52.51.246.115:/home/ubuntu/projects/deepdetect/models/mirrorshot_3/
+
+echo "---> sync mirrorshot #4"
+rsync -arvu --delete /data/projects/deepdetect_u1/models/mirrorshot_f1/ ubuntu@52.51.246.115:/home/ubuntu/projects/deepdetect/models/mirrorshot_4/
+
+echo "---> sync mirrorshot #5"
+rsync -arvu --delete /data/projects/deepdetect_u1/models/mirrorshot_f1/ ubuntu@52.51.246.115:/home/ubuntu/projects/deepdetect/models/mirrorshot_5/
+
+echo "---> sync mirrorshot #reanalyze"
+rsync -arvu --delete /data/projects/deepdetect_u1/models/mirrorshot_f1/ ubuntu@52.51.246.115:/home/ubuntu/projects/deepdetect/models/mirrorshot_reanalyze/
+
+echo -e "\nsync fine-tuned mirrorshot f1 done! \n"

data/lib/configuration.rb

@@ -0,0 +1,34 @@
+require "deepdetect_ruby/version"
+
+module DeepdetectRuby
+  module Configuration
+    VALID_CONFIG_KEYS = [:host, :host_train, :deepdetect_path, :model_path, :debug, :servers, :is_scaling].freeze
+    DEFAULT_HOST = "http://127.0.0.1:8080"
+    DEFAULT_HOST_TRAIN = "http://127.0.0.1:8080"
+
+    attr_accessor *VALID_CONFIG_KEYS
+
+    def self.extended(base)
+      base.reset
+    end
+
+    def reset
+      self.host = DEFAULT_HOST
+      self.host_train = DEFAULT_HOST_TRAIN
+      self.deepdetect_path = ""
+      self.model_path = ""
+      self.debug = true
+      self.servers = DEFAULT_HOST
+      self.is_scaling = false
+    end
+
+    def configure
+      yield self
+    end
+
+    def options
+      Hash[* VALID_CONFIG_KEYS.map { |key| [key, send(key)] }.flatten]
+    end
+
+  end
+end

data/lib/dede_server.rb

@@ -0,0 +1,24 @@
+module DeepdetectRuby
+  class DedeServer
+    def self.get_server(server_index = 1)
+      debug = DeepdetectRuby.options[:debug]
+      if DeepdetectRuby.options[:is_scaling]
+        servers = DeepdetectRuby.options[:servers].split(",").map{|s| s.strip}
+        puts "\n[get_server] server #{server_index} is running. We have #{servers.count} server working together. \n" if debug
+        if !servers.nil? && servers.present?
+          dede_index = server_index - 1
+          if dede_index >= 0
+            return servers[dede_index]
+          else
+            puts "\n[DedeServer] server_index must be greater than 0 \n" if debug
+          end
+        else
+          puts "\n[DedeServer] hosts must not be empty \n" if debug
+        end
+      else
+        puts "\n[get_server] single server is running \n" if debug
+        return DeepdetectRuby.options[:host]
+      end
+    end
+  end
+end

data/lib/deep_symbolize.rb

@@ -0,0 +1,60 @@
+# Symbolizes all of hash's keys and subkeys.
+# Also allows for custom pre-processing of keys (e.g. downcasing, etc)
+# if the block is given:
+#
+# somehash.deep_symbolize { |key| key.downcase }
+#
+# Usage: either include it into global Hash class to make it available to
+#        to all hashes, or extend only your own hash objects with this
+#        module.
+#        E.g.:
+#        1) class Hash; include DeepSymbolizable; end
+#        2) myhash.extend DeepSymbolizable
+
+module DeepSymbolizable
+  def deep_symbolize(&block)
+    method = self.class.to_s.downcase.to_sym
+    syms = DeepSymbolizable::Symbolizers
+    syms.respond_to?(method) ? syms.send(method, self, &block) : self
+  end
+
+  module Symbolizers
+    extend self
+
+    # the primary method - symbolizes keys of the given hash,
+    # preprocessing them with a block if one was given, and recursively
+    # going into all nested enumerables
+    def hash(hash, &block)
+      hash.inject({}) do |result, (key, value)|
+        # Recursively deep-symbolize subhashes
+        value = _recurse_(value, &block)
+
+        # Pre-process the key with a block if it was given
+        key = yield key if block_given?
+        # Symbolize the key string if it responds to to_sym
+        sym_key = key.to_sym rescue key
+
+        # write it back into the result and return the updated hash
+        result[sym_key] = value
+        result
+      end
+    end
+
+    # walking over arrays and symbolizing all nested elements
+    def array(ary, &block)
+      ary.map { |v| _recurse_(v, &block) }
+    end
+
+    # handling recursion - any Enumerable elements (except String)
+    # is being extended with the module, and then symbolized
+    def _recurse_(value, &block)
+      if value.is_a?(Enumerable) && !value.is_a?(String)
+        # support for a use case without extended core Hash
+        value.extend DeepSymbolizable unless value.class.include?(DeepSymbolizable)
+        value = value.deep_symbolize(&block)
+      end
+      value
+    end
+  end
+
+end

data/lib/deepdetect_ruby/version.rb

@@ -0,0 +1,3 @@
+module DeepdetectRuby
+  VERSION = "1.0.1"
+end

data/lib/deepdetect_ruby.rb

@@ -0,0 +1,13 @@
+require "deepdetect_ruby/version"
+require "configuration"
+require "deep_symbolize"
+require 'httparty'
+require 'dede_server'
+require "info"
+require "service"
+require "train"
+require "predict"
+
+module DeepdetectRuby
+  extend Configuration
+end

data/lib/info.rb

@@ -0,0 +1,42 @@
+module DeepdetectRuby
+  def self.info(options = {})
+    begin
+      server_index = options[:server_index] || 1
+      dede_host = DeepdetectRuby::DedeServer.get_server(server_index)
+      info_url = "#{dede_host}/info"
+      dede_info = HTTParty.get(info_url)
+      info = JSON.parse(dede_info.to_json)
+      info.extend DeepSymbolizable
+      return info.deep_symbolize
+    rescue Exception => e
+      puts "\n[DeepdetectRuby Info]. #{e.to_s} \n"
+      return nil
+    end
+  end
+
+  def self.status(options = {})
+    info(options)
+  end
+
+  def self.get_server(server_index = 1)
+    debug = DeepdetectRuby.options[:debug]
+    if DeepdetectRuby.options[:is_scaling]
+      servers = DeepdetectRuby.options[:servers].split(",").map{|s| s.strip}
+      puts "\n[get_server] server #{server_index} is running. We have #{servers.count} server working together. \n" if debug
+      if !servers.nil? && servers.present?
+        dede_index = server_index - 1
+        if dede_index >= 0
+          return servers[dede_index]
+        else
+          puts "\n[DedeServer] server_index must be greater than 0 \n" if debug
+        end
+      else
+        puts "\n[DedeServer] hosts must not be empty \n" if debug
+      end
+    else
+      puts "\n[get_server] single server is running \n" if debug
+      return DeepdetectRuby.options[:host]
+    end
+  end
+
+end

data/lib/predict.rb

@@ -0,0 +1,50 @@
+module DeepdetectRuby
+  class Predict
+    def self.predict(options = {})
+      begin
+        debug = DeepdetectRuby.options[:debug]
+        server_index = options[:server_index] || 1
+        dede_host = DeepdetectRuby::DedeServer.get_server(server_index)
+        dede_api_url = "#{dede_host}/predict"
+        puts "\n-------------> Starting to predict an image #{dede_api_url}....\n" if debug
+        best = options[:best] || 2
+        gpu = options[:gpu] || true
+        width = options[:width] || 224
+        height = options[:height] || 224
+
+        data = {
+          "service" => "#{options[:service]}",
+          "parameters" => {
+            "input" => {
+              "width" => width,
+              "height" => height
+            },
+            "output" => {
+              "best" => best
+            },
+            "mllib" => { "gpu" => gpu }
+          },
+          "data" => [
+            "#{options[:image_url]}"
+          ]
+        }
+        if !options[:template].nil?
+          data["parameters"]["output"]["template"] = options[:template]
+        end
+
+        puts "\nData: #{data.to_json} \n" if debug
+
+        dede_body = {
+          :body => data.to_json,
+          :headers => { "Content-Type" => 'application/json' }
+        }
+        object = HTTParty.post(dede_api_url, dede_body)
+        predict_info = JSON.parse(object.body)
+        predict_info.extend DeepSymbolizable
+        return predict_info.deep_symbolize
+      rescue Exception => e
+        puts "\n[DeepdetectRuby Predict - predict]. #{e.to_s} \n"
+      end
+    end # end predict
+  end
+end

data/lib/service.rb

@@ -0,0 +1,228 @@
+module DeepdetectRuby
+  class Service
+    def self.create(options = {})
+      begin
+        debug = DeepdetectRuby.options[:debug]
+        server_index = options[:server_index] || 1
+        dede_host = DeepdetectRuby::DedeServer.get_server(server_index)
+        dede_api_url = "#{dede_host}/services/#{options[:name]}"
+        puts "\n-------------> Starting to create DeepDetect service #{dede_api_url}....\n" if debug
+
+        description = "#{options[:name]} classification service"
+        options[:mllib] = !options[:mllib].nil? ? options[:mllib] : "caffe"
+        options[:description] = !options[:description].nil? ? options[:description] : description
+        options[:type] = !options[:type].nil? ? options[:type] : "supervised"
+        options[:connector] = !options[:connector].nil? ? options[:connector] : "image"
+        options[:width] = !options[:width].nil? ? options[:width] : 224
+        options[:height] = !options[:height].nil? ? options[:height] : 224
+        options[:nclasses] = !options[:nclasses].nil? ? options[:nclasses] : 2
+        options[:gpu] = !options[:gpu].nil? ? options[:gpu] : true
+
+        # for finetuning
+        options[:finetuning] = !options[:finetuning].nil? ? options[:finetuning] : false
+        options[:template] = !options[:template].nil? ? options[:template] : "googlenet"
+        options[:weights] = !options[:weights].nil? ? options[:weights] : nil
+        options[:rotate] = !options[:rotate].nil? ? options[:rotate] : false
+        options[:mirror] = !options[:mirror].nil? ? options[:mirror] : false
+        options[:mirror] = options[:finetuning]
+
+         if !DeepdetectRuby.options[:model_path].nil? && !DeepdetectRuby.options[:model_path].empty?
+          options[:repository] = "#{DeepdetectRuby.options[:model_path]}/#{options[:name]}"
+        end
+
+        # for different model name
+        if !options[:model_name].nil? && !options[:model_name].empty?
+          options[:repository] = "#{DeepdetectRuby.options[:model_path]}/#{options[:model_name]}"
+        end
+
+        data = {
+          "mllib" => "#{options[:mllib]}",
+          "description" => "#{options[:description]}",
+          "type" => "#{options[:type]}",
+          "parameters" => {
+            "input" => {
+              "connector" => "#{options[:connector]}",
+              "height" => options[:height],
+              "width" => options[:width]
+            },
+            "mllib" => {
+              "nclasses" => options[:nclasses],
+              "gpu" => options[:gpu]
+            }
+          },
+          "model" => {
+            "repository" => "#{options[:repository]}"
+          }
+        }
+
+        if options[:finetuning] && !options[:weights].nil?
+          data = {
+                    "mllib" => "#{options[:mllib]}",
+                    "description" => "#{options[:description]}",
+                    "type" => "#{options[:type]}",
+                    "parameters" => {
+                      "input" => {
+                        "connector" => "#{options[:connector]}",
+                        "height" => options[:height],
+                        "width" => options[:width]
+                      },
+                      "mllib" => {
+                        "nclasses" => options[:nclasses],
+                        "gpu" => options[:gpu],
+                        "weights" => options[:weights],
+                        "finetuning" => options[:finetuning],
+                        "rotate" => options[:rotate],
+                        "mirror" => options[:mirror]
+                      }
+                    },
+                    "model" => {
+                      "repository" => "#{options[:repository]}"
+                    }
+                  }
+        end
+
+        puts "\nparams data: #{data.to_json} \n" if debug
+        dede_body = {
+          :body => data.to_json,
+          :headers => { "Content-Type" => 'application/json' }
+        }
+        object = HTTParty.put(dede_api_url, dede_body)
+        object_info = JSON.parse(object.body)
+        object_info.extend DeepSymbolizable
+        return object_info.deep_symbolize
+      rescue Exception => e
+        puts "\n[DeepdetectRuby Service]. #{e.to_s} \n"
+      end
+
+    end
+
+    def self.get_info(service_name = "", server_index = 1)
+      begin
+        debug = DeepdetectRuby.options[:debug]
+        dede_host = DeepdetectRuby::DedeServer.get_server(server_index)
+        dede_api_url = "#{dede_host}/services/#{service_name}"
+        puts "\n-------------> Starting to get information DeepDetect service #{dede_api_url}....\n" if debug
+        object = HTTParty.get(dede_api_url)
+        object_info = JSON.parse(object.body)
+        object_info.extend DeepSymbolizable
+        return object_info.deep_symbolize
+      rescue Exception => e
+        puts "\n[DeepdetectRuby Service get information]. #{e.to_s} \n"
+      end
+    end
+
+    def self.delete(service_name = "", server_index = 1)
+      begin
+        debug = DeepdetectRuby.options[:debug]
+        dede_host = DeepdetectRuby::DedeServer.get_server(server_index)
+        dede_api_url = "#{dede_host}/services/#{service_name}"
+        puts "\n-------------> Starting to delete DeepDetect service #{dede_api_url}....\n" if debug
+        object = HTTParty.delete(dede_api_url)
+        object_info = JSON.parse(object.body)
+        object_info.extend DeepSymbolizable
+        return object_info.deep_symbolize
+      rescue Exception => e
+        puts "\n[DeepdetectRuby delete service]. #{e.to_s} \n"
+      end
+    end
+
+    def self.load_all_models(options = {})
+      debug = DeepdetectRuby.options[:debug]
+      server_index = options[:server_index] || 1
+      dede_host = DeepdetectRuby::DedeServer.get_server(server_index)
+      puts "\n-----------> Starting to load_all_models..." if debug
+      model_path = DeepdetectRuby.options[:model_path]
+      dirs = Dir.glob("#{model_path}/*").select{ |e| File.directory? e }
+      dirs.each_with_index do |dir, index|
+        model_name = File.basename(dir)
+        dede_opts = { repository: dir, name: model_name, server_index: server_index }
+        DeepdetectRuby::Service.create(dede_opts)
+        puts "Index: #{index} - model name: #{model_name} in #{dir}" if debug
+      end
+      puts "\n-----------> End to load_all_models..." if debug
+    end
+
+    def self.reload_models(options = {})
+      debug = DeepdetectRuby.options[:debug]
+      server_index = options[:server_index] || 1
+      dede_host = DeepdetectRuby::DedeServer.get_server(server_index)
+      puts "\n-----------> Starting to reload models..." if debug
+      model_path = DeepdetectRuby.options[:model_path]
+      dirs = Dir.glob("#{model_path}/*").select{ |e| File.directory? e }
+      dirs.each_with_index do |dir, index|
+        model_name = File.basename(dir)
+        DeepdetectRuby::Service.delete(model_name, server_index)
+      end
+      DeepdetectRuby::Service.load_all_models(server_index)
+      puts "\n-----------> End to reload models..." if debug
+    end
+
+    def self.list_models
+      debug = DeepdetectRuby.options[:debug]
+      puts "\n-----------> Starting to reload models..." if debug
+      model_path = DeepdetectRuby.options[:model_path]
+      dirs = Dir.glob("#{model_path}/*").select{ |e| File.directory? e }
+      list = []
+      dirs.each_with_index do |dir, index|
+        model_name = File.basename(dir)
+        list << model_name
+      end
+      puts "\nList models: #{list.to_json}" if debug
+      return list
+    end
+
+    def self.list_models_running(options = {})
+      dede = DeepdetectRuby.info(options)
+      services = dede[:head][:services]
+      return services
+    end
+
+    def self.finetuning(options = {})
+      puts "\n-----> Starting to create service and train for finetuning \n"
+      server_index = options[:server_index] || 1
+      dede_host = DeepdetectRuby::DedeServer.get_server(server_index)
+      options[:name] = !options[:name].nil? ? options[:name] : ""
+      # weights is called the caffe filename model, example: model_iter_16368.caffemodel
+      options[:weights] = !options[:weights].nil? ? options[:weights] : ""
+      options[:repository] = !options[:repository].nil? ? options[:repository] : ""
+      options[:gpu] = !options[:gpu].nil? ? options[:gpu] : true
+      options[:iterations] = !options[:iterations].nil? ? options[:iterations] : 10000
+      options[:test_interval] = !options[:test_interval].nil? ? options[:test_interval] : 500
+      options[:nclasses] = !options[:nclasses].nil? ? options[:nclasses] : 2
+      options[:measure_index] = !options[:measure_index].nil? ? options[:measure_index] : 1
+      options[:batch_size] = !options[:batch_size].nil? ? options[:batch_size] : 32
+      options[:test_batch_size] = !options[:test_batch_size].nil? ? options[:test_batch_size] : 32
+
+      if !options[:name].empty? && !options[:weights].empty? && !options[:repository].empty?
+        # Create a service for finetuning
+        service_options = { name: "#{options[:name]}",
+                            finetuning: true,
+                            weights: "#{options[:weights]}",
+                            nclasses: options[:nclasses],
+                            server_index: server_index
+                          }
+        DeepdetectRuby::Service.create(service_options)
+
+        puts "\nWaiting to start function fine tune..."
+        sleep(3)
+
+        # Train a new updated service for finetuning
+        train_options = { service: "#{options[:name]}",
+                          repository: "#{options[:repository]}",
+                          gpu: true,
+                          iterations: options[:iterations],
+                          test_interval: options[:test_interval],
+                          finetuning: true,
+                          measure_index: options[:measure_index],
+                          batch_size: options[:batch_size],
+                          test_batch_size: options[:test_batch_size],
+                          server_index: server_index
+                        }
+        DeepdetectRuby::Train.launch(train_options)
+      else
+        puts "\nMissing options :name or :weights or :repository. Please check. :) \n"
+      end
+    end
+
+  end
+end