menoh 1.0.0

data/ext/menoh_native/menoh_ruby.h

@@ -0,0 +1,7 @@
+#ifndef RUNX_H
+#define RUNX_H 1
+
+#include <menoh/menoh.h>
+#include <ruby.h>
+
+#endif /* RUNX_H */

data/lib/menoh.rb

@@ -0,0 +1,87 @@
+require 'menoh/version'
+require 'menoh/menoh_native'
+
+module Menoh
+  class Menoh
+    def initialize(file)
+      if !File.exist?(file)
+        raise "No such file : #{file}"
+      end
+
+      native_init file
+      yield self if block_given?
+    end
+
+    def make_model(option)
+      if option[:backend].nil?
+        raise "Required ':backend' : #{option[:backend]}"
+      end
+      model = MenohModel.new self, option
+      yield model if block_given?
+      model
+    end
+  end
+end
+
+def transpose(buffer, shape)
+  sliced_buffer = buffer.each_slice(buffer.length / shape[0]).to_a
+  if shape.length > 2
+    next_shape = shape.slice(1, a.length)
+    sliced_buffer = sliced_buffer.map { |buf| transpose buf, next_shape }
+  end
+  sliced_buffer
+end
+
+module Menoh
+  class MenohModel
+    def initialize(menoh, option)
+      if option[:input_layers].nil? || option[:input_layers].empty?
+        raise "Required ':input_layers'"
+      end
+      raise "Required ':input_layers'" unless option[:input_layers].instance_of?(Array)
+      option[:input_layers].each_with_index do |input_layer, i|
+        raise 'Invalid option : input_layers' unless input_layer.instance_of?(Hash)
+        raise "Invalid name for input_layer[#{i}]" unless input_layer[:name].instance_of?(String)
+        raise "Invalid dims for input_layer[#{i}]" unless input_layer[:dims].instance_of?(Array)
+      end
+      if option[:output_layers].nil? || option[:output_layers].empty?
+        raise "Invalid ':output_layers'"
+      end
+      native_init menoh, option
+      @option = option
+      yield self if block_given?
+    end
+
+    def run(dataset)
+      raise 'Invalid dataset' if !dataset.instance_of?(Array) || dataset.empty?
+      if dataset.length != @option[:input_layers].length
+        raise "Invalid input num: expected==#{@option[:input_layers].length} actual==#{dataset.length}"
+      end
+      dataset_for_native = []
+      dataset.each do |input|
+        if !input[:data].instance_of?(Array) || input[:data].empty?
+          raise "Invalid dataset for layer #{input[:name]}"
+        end
+        target_layer = @option[:input_layers].find { |item| item[:name] == input[:name] }
+        expected_data_length = target_layer[:dims].inject(:*)
+        if input[:data].length != expected_data_length
+          raise "Invalid data length: expected==#{expected_data_length} actual==#{input[:data].length}"
+        end
+        dataset_for_native << input[:data]
+      end
+
+      # run
+      results = native_run dataset_for_native
+
+      # reshape result
+      results.map do |raw|
+        buffer = raw[:data]
+        shape = raw[:shape]
+        raw[:data] = transpose buffer, shape
+      end
+
+      yield results if block_given?
+      results
+    end
+  end
+end

data/lib/menoh/version.rb

@@ -0,0 +1,3 @@
+module Menoh
+  VERSION = '1.0.0'.freeze
+end

data/menoh.gemspec

@@ -0,0 +1,29 @@
+lib = File.expand_path('lib', __dir__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'menoh/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'menoh'
+  spec.version       = Menoh::VERSION
+  spec.authors       = ['Kunihiko MIYOSHI']
+  spec.email         = ['miyoshik@preferred.jp']
+
+  spec.summary       = "Ruby binding of ONNX runtime engine 'Menoh'"
+  spec.description   = "Ruby binding of ONNX runtime engine 'Menoh'"
+  spec.homepage      = 'https://github.com/pfnet-research/menoh-ruby'
+  spec.license       = 'MIT'
+
+  spec.files         = `git ls-files -z`.split("\x0").reject do |f|
+    f.match(%r{^(test|spec|features)/})
+  end
+  spec.bindir        = 'exe'
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ['lib']
+  spec.extensions    = ['ext/menoh_native/extconf.rb']
+
+  spec.add_development_dependency 'bundler', '~> 1.16'
+  spec.add_development_dependency 'minitest', '~> 5.0'
+  spec.add_development_dependency 'pry'
+  spec.add_development_dependency 'rake', '~> 10.0'
+  spec.add_development_dependency 'rake-compiler'
+end

metadata

@@ -0,0 +1,146 @@
+--- !ruby/object:Gem::Specification
+name: menoh
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Kunihiko MIYOSHI
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2018-07-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.16'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.16'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rake-compiler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Ruby binding of ONNX runtime engine 'Menoh'
+email:
+- miyoshik@preferred.jp
+executables: []
+extensions:
+- ext/menoh_native/extconf.rb
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- CODE_OF_CONDUCT.md
+- Dockerfile
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- Vagrantfile
+- bin/console
+- bin/setup
+- docs/tutorial.md
+- example/data/0.png
+- example/data/1.png
+- example/data/2.png
+- example/data/3.png
+- example/data/4.png
+- example/data/5.png
+- example/data/6.png
+- example/data/7.png
+- example/data/8.png
+- example/data/9.png
+- example/data/mnist.onnx
+- example/example_mnist.rb
+- example/example_mnist_with_block.rb
+- example/example_vgg16.rb
+- ext/menoh_native/extconf.rb
+- ext/menoh_native/menoh_ruby.c
+- ext/menoh_native/menoh_ruby.h
+- lib/menoh.rb
+- lib/menoh/version.rb
+- menoh.gemspec
+homepage: https://github.com/pfnet-research/menoh-ruby
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.2.1
+signing_key: 
+specification_version: 4
+summary: Ruby binding of ONNX runtime engine 'Menoh'
+test_files: []