embulk-filter-google_vision_api 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 129cfa811e9b9b1a2e94004d2c32fc8c7008cc99
+  data.tar.gz: be9ea20997ac41ee0126b5b137c175defbeb2e7d
+SHA512:
+  metadata.gz: 0177f5ffe7ae5f07012ae1a934ba26b2bcbf87ef4dc3ea24946af7a6f3c33a1458fdde84e6066937ec5dde9d5f49c87e74d1b59aab1c1bd6ebebe17897c469f1
+  data.tar.gz: a11a2816eea10df168d759783f5c9d5b2b66071b178cb540bc153b870ccd777ce94f07d228ea63ad427e72aec3e162623da35b155bf9ad5d8631fdb3e1744b91

data/.gitignore

@@ -0,0 +1,5 @@
+*~
+/pkg/
+/tmp/
+/.bundle/
+/Gemfile.lock

data/.ruby-version

@@ -0,0 +1 @@
+jruby-9.1.5.0

data/Gemfile

@@ -0,0 +1,2 @@
+source 'https://rubygems.org/'
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,73 @@
+# Google Vision Api filter plugin for Embulk
+
+Google Vision Api filter plugin for Embulk.
+
+Very easy image recognition.
+
+## Overview
+
+* **Plugin type**: filter
+
+## Configuration
+
+- **out_key_name**: out_key_name (string)
+- **image_path_key_name**: image_path_key_name (string)
+- **features**: features (array)
+  - **type**: FACE_DETECTION or LANDMARK_DETECTION or LOGO_DETECTION or LABEL_DETECTION or TEXT_DETECTION or SAFE_SEARCH_DETECTION or IMAGE_PROPERTIES (string)
+  - **maxResults**: maxResults (integer)
+- **delay**: delay (integer, default: 0)
+- **image_num_per_request**: image_num_per_request (integer, default: 16)
+- **google_api_key**: google_api_key (string, default: ENV['GOOGLE_API_KEY'])
+
+## Example
+
+### input
+```yaml
+    - { image_path: 'http://www.embulk.org/docs/_images/embulk-logo.png' }
+```
+
+* respond localfile path and http URI.
+
+### setting
+```yaml
+filters:
+  - type: google_vision_api
+    image_path_key_name: image_path
+    out_key_name: image_info
+    image_num_per_request: 5
+    features: 
+      - {type: FACE_DETECTION, "maxResults":5 }
+      - {type: LANDMARK_DETECTION, "maxResults":5 }
+      - {type: LOGO_DETECTION, "maxResults":5 }
+      - {type: LABEL_DETECTION, "maxResults":5 }
+      - {type: TEXT_DETECTION, "maxResults":5 }
+      - {type: SAFE_SEARCH_DETECTION, "maxResults":5 }
+      - {type: IMAGE_PROPERTIES, "maxResults":5 }
+```
+
+### output
+```
+image_path (string) : http://www.embulk.org/docs/_images/embulk-logo.png
+image_info (  json) : {"labelAnnotations":[{"mid":"/m/0dwx7","description":"logo","score":0.86478204},{"mid":"/m/03gq5hm","description":"font","score":0.8472268},{"mid":"/m/0215n","description":"cartoon","score":0.82311255},{"mid":"/m/03g09t","description":"clip art","score":0.69382942},{"mid":"/m/01cd9","description":"brand","score":0.59691668}],"textAnnotations":[{"locale":"id","description":"embulk\n","boundingPoly":{"vertices":[{"x":67,"y":475},{"x":812,"y":475},{"x":812,"y":629},{"x":67,"y":629}]}},{"description":"embulk","boundingPoly":{"vertices":[{"x":68,"y":475},{"x":813,"y":475},{"x":813,"y":629},{"x":68,"y":629}]}}],"safeSearchAnnotation":{"adult":"VERY_UNLIKELY","spoof":"UNLIKELY","medical":"UNLIKELY","violence":"VERY_UNLIKELY"},"imagePropertiesAnnotation":{"dominantColors":{"colors":[{"color":{"red":231,"green":59,"blue":11},"score":0.26240975,"pixelFraction":0.029842343},{"color":{"red":243,"green":176,"blue":124},"score":0.0057866224,"pixelFraction":0.010698198},{"color":{"red":252,"green":250,"blue":248},"score":0.0025187095,"pixelFraction":0.6255005},{"color":{"red":232,"green":79,"blue":16},"score":0.17388013,"pixelFraction":0.023773775},{"color":{"red":225,"green":75,"blue":27},"score":0.14469221,"pixelFraction":0.034909911},{"color":{"red":222,"green":52,"blue":17},"score":0.13995738,"pixelFraction":0.018768769},{"color":{"red":243,"green":106,"blue":31},"score":0.090499125,"pixelFraction":0.025525525},{"color":{"red":243,"green":128,"blue":38},"score":0.064238794,"pixelFraction":0.050425425},{"color":{"red":242,"green":104,"blue":47},"score":0.033798043,"pixelFraction":0.014451952},{"color":{"red":238,"green":131,"blue":65},"score":0.027836611,"pixelFraction":0.010948448}]}}}
+```
+
+## Vision API Limits
+
+| Type of Limit | Usage Limit |
+|:-----------|------------:|
+| MB per image |  4 MB |
+| MB per request |  8 MB |
+| Requests per second | 10 |
+| Requests per feature per day | 700,000 |
+| Requests per feature per month | 20,000,000 |
+| Images per second | 8 |
+| Images per request |  16 |
+
+see. [Usage Limits  \|  Google Cloud Vision API  \|  Google Cloud Platform](https://cloud.google.com/vision/limits)
+
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

@@ -0,0 +1,3 @@
+require "bundler/gem_tasks"
+
+task default: :build

data/embulk-filter-google_vision_api.gemspec

@@ -0,0 +1,20 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-filter-google_vision_api"
+  spec.version       = "0.1.0"
+  spec.authors       = ["toyama0919"]
+  spec.summary       = "Google Vision Api filter plugin for Embulk"
+  spec.description   = "Google Vision Api filter plugin for Embulk. Very easy image recognition."
+  spec.email         = ["toyama0919@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/toyama0919/embulk-filter-google_vision_api"
+
+  spec.files         = `git ls-files`.split("\n") + Dir["classpath/*.jar"]
+  spec.test_files    = spec.files.grep(%r{^(test|spec)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency 'jruby-openssl'
+  spec.add_development_dependency 'embulk', ['>= 0.8.15']
+  spec.add_development_dependency 'bundler', ['>= 1.10.6']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/lib/embulk/filter/google_vision_api.rb

@@ -0,0 +1,102 @@
+require "json"
+require "net/http"
+require "uri"
+require "openssl"
+require "base64"
+require "pp"
+
+module Embulk
+  module Filter
+
+    class GoogleVisionApi < FilterPlugin
+      Plugin.register_filter("google_vision_api", self)
+      ENDPOINT_PREFIX = "https://vision.googleapis.com/v1/images:annotate"
+
+      def self.transaction(config, in_schema, &control)
+        task = {
+          "out_key_name" => config.param("out_key_name", :string),
+          "image_path_key_name" => config.param("image_path_key_name", :string),
+          "features" => config.param("features", :array),
+          "delay" => config.param("delay", :integer, default: 0),
+          "image_num_per_request" => config.param("image_num_per_request", :integer, default: 16),
+          "google_api_key" => config.param("google_api_key", :string, default: ENV['GOOGLE_API_KEY']),
+        }
+
+        add_columns = [
+          Column.new(nil, task["out_key_name"], :json)
+        ]
+
+        out_columns = in_schema + add_columns
+
+        yield(task, out_columns)
+      end
+
+      def init
+        @uri = URI.parse("#{ENDPOINT_PREFIX}?key=#{task['google_api_key']}")
+        @http = Net::HTTP.new(@uri.host, @uri.port)
+        @http.use_ssl = true
+        @http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+        @post = Net::HTTP::Post.new(@uri.request_uri, initheader = {'Content-Type' =>'application/json'})
+        @image_path_key_name = task['image_path_key_name']
+        @out_key_name = task['out_key_name']
+        @delay = task['delay']
+        @image_num_per_request = task['image_num_per_request']
+        @features = task['features']
+      end
+
+      def close
+      end
+
+      def add(page)
+        record_groups = page.map { |record|
+          Hash[in_schema.names.zip(record)]
+        }.each_slice(@image_num_per_request).to_a
+
+        record_groups.each do |records|
+          requests = []
+          records.each do |record|
+            request = {
+              image: {},
+              features: @features
+            }
+            image_body = get_image_body(record)
+            request[:image][:content] = Base64.encode64(image_body)
+            requests << request
+          end
+          body = {
+            requests: requests
+          }
+          @post.body = body.to_json
+          Embulk.logger.debug "request body => #{@post.body}"
+
+          response_hash = {}
+          @http.start do |h|
+            response = h.request(@post)
+            response_hash = JSON.parse(response.body)
+          end
+          records.each_with_index do |record, i|
+            recognized = response_hash['responses'][i]
+            Embulk.logger.warn "Error image => [#{record[@image_path_key_name]}] #{recognized}" if recognized.key?("error")
+            page_builder.add(record.values + [recognized])
+          end
+
+          sleep @delay
+        end
+      end
+
+      def finish
+        page_builder.finish
+      end
+
+      private
+      def get_image_body(record)
+        image_path = record[@image_path_key_name]
+        if image_path =~ /https?\:\/\//
+          Net::HTTP.get_response(URI.parse(image_path)).body rescue ""
+        else
+          File.read(image_path) rescue ""
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,108 @@
+--- !ruby/object:Gem::Specification
+name: embulk-filter-google_vision_api
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- toyama0919
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2017-01-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: jruby-openssl
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.15
+  name: embulk
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.15
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.10.6
+  name: bundler
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.10.6
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  name: rake
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: Google Vision Api filter plugin for Embulk. Very easy image recognition.
+email:
+- toyama0919@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".ruby-version"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-filter-google_vision_api.gemspec
+- lib/embulk/filter/google_vision_api.rb
+homepage: https://github.com/toyama0919/embulk-filter-google_vision_api
+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.6.6
+signing_key:
+specification_version: 4
+summary: Google Vision Api filter plugin for Embulk
+test_files: []