opencv-color 1.0.2 → 1.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 3d1ab524bb6fc4578ddff0366bf1211fa8aeb075
-  data.tar.gz: fd49bdf066ced28c96961ef4a9e2694d3f8ad4ce
+  metadata.gz: 3946cad363e3d99564e4160626524cc759f53827
+  data.tar.gz: 97cc778daa0740ea60e451a5f6a98e60bb1fd0d8
 SHA512:
-  metadata.gz: f99dfd480a7d83a3b9039e4d99fe0d096756daa70dcad9cda613126f137286ec4588c4a9d502fb3efc27b3aa165070516f730282bfd0beb45e51f17702a2c9be
-  data.tar.gz: 2ea7688d6a4e4a7501e5aa59cb0b9fa717c6eeb6ce33aa0be418ef6f87f96737145b20e785ed54326d4dc6ed2663636938a87dc7db8c33a840e3405516a69bc9
+  metadata.gz: fd4c2c695f82da3731016169fe6da72128b72231ecbb9ffa7a449f3771d9a03cc422360fd5f3789ef65d5d6f2b6c48e6ef3dafb3bce2246f8b42493c8ef4c50b
+  data.tar.gz: 5a47009144dc0164cf9ae034aee8290e03e83896f213602e2540c094f6c1cfa5112984305944af8e9c03741ac571100fdecbb5f09300111f0139da55a2e60d80

data/README.md

@@ -1,5 +1,7 @@
 # OpenCVColor
 
+[Built with :yellow_heart: and :coffee: in San Francisco](http://getmingle.io)
+
 OpenCV is a great library for writing Computer Vision software. However, OpenCV's HSV format is different than what you would expect! This gem trys to learn HSV color range from sample images for your OpenCV program to detect color.
 
 ## Installation

data/bin/opencv-color

@@ -7,6 +7,7 @@ require 'erb'
 #add lib for test environment
 if File.exists?(File.expand_path(File.dirname(__FILE__) + '/../lib/opencv-color.rb'))
   $LOAD_PATH.unshift File.expand_path(File.dirname(__FILE__) + '/../lib')
+  require 'rubygems'
 end
 
 begin
@@ -19,12 +20,12 @@ end
 require 'optparse'
 
 options = {
-  format: 'yaml'
+  format: 'table'
 }
 op = OptionParser.new do |opts|
   opts.banner = "Usage: opencv-color [options] <sample images directory path>"
 
-  opts.on("-f", "--format [FORMAT]", "Output format 'yaml' or 'objc', default is yaml") do |format|
+  opts.on("-f", "--format [FORMAT]", "Output format 'yaml', 'table' or 'objc', default is table") do |format|
     options[:format] = format
   end
 
@@ -44,11 +45,17 @@ unless File.directory?(sample_images_dir)
   exit(-1)
 end
 
-data = OpenCVColor.learn(sample_images_dir)
+data = OpenCVColor.learn(sample_images_dir).sort_by{|k,v|v[:low][0]}
 puts case options[:format]
 when 'objc'
   template = File.join(File.dirname(__FILE__), '..', 'lib', 'opencv-color', 'objc.erb')
   ERB.new(File.read(template)).result(binding)
-else
+when 'yaml'
   YAML.dump(data)
+when 'table'
+  require 'terminal-table'
+  rows = data.map {|k,v| [k, v[:low], v[:high], v[:mean].map(&:to_i), v[:sd].map(&:to_i)]}
+  table = Terminal::Table.new :headings => ["Color", "Low", "High", "Mean", "SD"], :rows => rows
+else
+  puts "Unknown format: #{options[:format].inspect}"
 end

data/color-samples.tb

@@ -0,0 +1,15 @@
++----------------+----------------+-----------------+----------------+-------------+
+| Color          | Low            | High            | Mean           | SD          |
++----------------+----------------+-----------------+----------------+-------------+
+| pink_0         | [0, 44, 203]   | [5, 75, 255]    | [1, 59, 232]   | [0, 4, 9]   |
+| yellow_dark_0  | [23, 211, 196] | [27, 255, 255]  | [24, 233, 229] | [0, 7, 10]  |
+| yellow_light_0 | [25, 57, 178]  | [31, 117, 255]  | [27, 87, 232]  | [0, 9, 18]  |
+| yellow_green_0 | [28, 87, 146]  | [33, 141, 255]  | [30, 114, 217] | [0, 8, 23]  |
+| green_0        | [28, 30, 182]  | [40, 179, 255]  | [33, 104, 222] | [1, 24, 13] |
+| yellow_0       | [29, 192, 208] | [33, 218, 255]  | [30, 205, 235] | [0, 4, 9]   |
+| green_dark_0   | [85, 137, 97]  | [92, 159, 117]  | [88, 147, 107] | [0, 3, 3]   |
+| blue_light_0   | [89, 35, 129]  | [97, 87, 255]   | [93, 60, 217]  | [1, 8, 29]  |
+| blue_0         | [95, 98, 143]  | [100, 153, 238] | [97, 125, 190] | [0, 9, 15]  |
+| purple_0       | [117, 7, 125]  | [130, 58, 194]  | [123, 32, 159] | [1, 8, 11]  |
+| pink_1         | [177, 49, 220] | [179, 68, 253]  | [178, 58, 236] | [0, 3, 5]   |
++----------------+----------------+-----------------+----------------+-------------+

data/lib/opencv-color.rb

@@ -23,55 +23,108 @@ module OpenCVColor
     end
   end
 
+  class Cluster
+    def initialize(color)
+      @h, @s, @v = [], [], []
+      @hs = 0
+      self << color
+    end
+
+    def <<(color)
+      @h << color[0]
+      @s << color[1]
+      @v << color[2]
+      @hs += color[0]
+    end
+
+    def distance(color)
+      (center - color[0]).abs
+    end
+
+    def center
+      @hs / @h.size
+    end
+
+    def colors
+      [@h, @s, @v]
+    end
+
+    def color_range
+      memo = {low: [], high: [], mean: [], sd: []}
+      colors.map(&:to_scale).each_with_index do |d, index|
+        sd = d.sd
+        mean = d.mean
+
+        memo[:low] << ([mean - 3 * sd, 0.0].max).floor
+        memo[:high] << ([mean + 3 * sd, max_value(index)].min).ceil
+        memo[:mean] << mean
+        memo[:sd] << sd
+      end
+      memo
+    end
+
+    def max_value(i)
+      # H => 0
+      i == 0 ? 179.0 : 255.0
+    end
+  end
+
   include OpenCV
 
   module_function
-  def samples(dir)
-    Dir["#{dir}/*"].map do |color|
-      [color, Dir["#{color}/*"]]
+  def normalize_colors(colors)
+    size = colors.map(&:size).min
+    colors.map do |cs|
+      cs.to_a.shuffle.first(size)
     end
   end
 
-  def learn(dir)
-    Hash[samples(dir).map do |color_dir, files|
-      group = [[], [], []]
-      files.map(&method(:load_image_colors)).each do |colors|
-        colors.each do |c|
-          3.times do |i|
-            group[i] << c[i]
-          end
+  def cluster_colors(colors, max_distance=20)
+    clusters = []
+    colors.each do |cs|
+      cs.each do |color|
+        cluster = clusters.find do |cluster|
+          cluster.distance(color) < max_distance
+        end
+
+        if cluster
+          cluster << color
+        else
+          clusters << Cluster.new(color)
         end
       end
-      range = group.map(&:to_scale).each_with_index.inject({low: [], high: [], mean: [], sd: []}) do |memo, d|
-        g, index = d
-        sd = g.sd
-        mean = g.mean
-        memo[:low] << ([mean - 3 * sd, min_value].max).floor
-        memo[:high] << ([mean + 3 * sd, max_value(index)].min).ceil
-        memo[:mean] << mean
-        memo[:sd] << sd
-        memo
-      end
-      [color_name(color_dir), range]
-    end]
+    end
+    clusters
   end
 
-  def color_name(path)
-    File.basename(path).downcase.gsub(/[^a-z_]/, '_')
+  def samples(dir)
+    Dir["#{dir}/*"].map do |color|
+      [color, Dir["#{color}/*"]]
+    end
   end
 
-  def min_value
-    0
+  def learn(dir)
+    ret = samples(dir).inject({}) do |memo, cd|
+      color_dir, files = cd
+      colors = files.map(&method(:load_image_colors))
+      cluster_colors(normalize_colors(colors)).each_with_index do |cluster, i|
+        memo[color_name(color_dir, i)] = cluster.color_range
+      end
+      memo
+    end
+    Hash[ret]
   end
 
-  def max_value(i)
-    # H => 0
-    i == 0 ? 179 : 255
+  def color_name(path, i)
+    "#{File.basename(path).downcase.gsub(/[^a-z_]/, '_')}_#{i}"
   end
 
   def load_image_colors(file)
     img = IplImage.load(file, OpenCV::CV_LOAD_IMAGE_ANYCOLOR | OpenCV::CV_LOAD_IMAGE_ANYDEPTH)
-    Colors.new(img.smooth(:median, 3).BGR2HSV)
+    colors(img.BGR2HSV)
   end
 
+  def colors(img)
+    Colors.new(img)
+  end
 end

data/lib/opencv-color/version.rb

@@ -1,3 +1,3 @@
 module OpenCVColor
-  VERSION = "1.0.2"
+  VERSION = "1.0.3"
 end

data/opencv-color.gemspec

@@ -20,6 +20,7 @@ Gem::Specification.new do |spec|
 
   spec.add_dependency('ruby-opencv', "~> 0.0.13")
   spec.add_dependency('statsample', "~> 1.4.0")
+  spec.add_dependency('terminal-table', "~> 1.4.5")
   spec.add_development_dependency "bundler", "~> 1.7"
   spec.add_development_dependency "rake", "~> 10.0"
 end

data/test/opencv_color_test.rb

@@ -0,0 +1,95 @@
+require 'minitest/autorun'
+
+require "opencv-color"
+
+class TestOpencvColor < MiniTest::Test
+
+  class Image
+    def initialize(data)
+      @data = data
+    end
+
+    def width
+      height == 0 ? 0 : @data[0].size
+    end
+
+    def height
+      @data.size
+    end
+
+    def [](i, j)
+      @data[i][j]
+    end
+  end
+
+  def test_cluster_colors
+    image1 = Image.new([
+      [[0,1,1], [1,1,1]],
+      [[189,1,1], [190,1,1]]
+    ])
+
+    clusters = OpenCVColor.cluster_colors([OpenCVColor.colors(image1)])
+    assert_equal 2, clusters.size
+    assert_equal [[0,1], [1,1],[1,1]], clusters[0].colors
+    assert_equal [[189,190], [1,1], [1,1]], clusters[1].colors
+
+    image2 = Image.new([
+      [[0,1,1], [1,1,1], [5,1,1], [111, 1, 1]],
+      [[3,1,1], [180,1,1], [100,1,1], [115,1,1]]
+    ])
+
+    clusters = OpenCVColor.cluster_colors([OpenCVColor.colors(image1), OpenCVColor.colors(image2)])
+
+    assert_equal 3, clusters.size
+    assert_equal [[0,1,0,1,5,3], [1,1,1,1,1,1], [1,1,1,1,1,1]], clusters[0].colors
+    assert_equal [[189,190,180], [1,1,1], [1,1,1]], clusters[1].colors
+    assert_equal [[111,100,115], [1,1,1], [1,1,1]], clusters[2].colors
+  end
+
+  def test_cluster_color_range
+    image1 = Image.new([
+      [[10,10,10], [12,12,12]],
+      [[150,150,150], [152,152,152]]
+    ])
+    clusters = OpenCVColor.cluster_colors([OpenCVColor.colors(image1)])
+    assert_equal [6,6,6], clusters[0].color_range[:low]
+    assert_equal [16,16,16], clusters[0].color_range[:high]
+    assert_equal [11,11,11], clusters[0].color_range[:mean]
+
+    assert_equal [146,146,146], clusters[1].color_range[:low]
+    assert_equal [156,156,156], clusters[1].color_range[:high]
+    assert_equal [151,151,151], clusters[1].color_range[:mean]
+  end
+
+  def test_cluster_color_range_min_and_max_values
+    image1 = Image.new([
+      [[0,0,0], [2,2,2]],
+      [[175,250,250], [179,254,254]]
+    ])
+    clusters = OpenCVColor.cluster_colors([OpenCVColor.colors(image1)])
+    assert_equal [0,0,0], clusters[0].color_range[:low]
+    assert_equal [6,6,6], clusters[0].color_range[:high]
+    assert_equal [1,1,1], clusters[0].color_range[:mean]
+
+    assert_equal [168,243,243], clusters[1].color_range[:low]
+    assert_equal [179,255,255], clusters[1].color_range[:high]
+    assert_equal [177, 252, 252], clusters[1].color_range[:mean]
+  end
+
+  def test_normalize_colors_by_min_image_size
+    image1 = Image.new([
+      [[0,0,0], [2,2,2]],
+      [[175,250,250], [179,254,254]]
+    ])
+    image2 = Image.new([
+      [[0,0,0]],
+      [[175,250,250]]
+    ])
+    image3 = Image.new([
+      [[0,0,0],[0,0,0],[0,0,0]],
+      [[175,250,250],[0,0,0],[0,0,0]]
+    ])
+    colors = OpenCVColor.normalize_colors([OpenCVColor.colors(image1), OpenCVColor.colors(image2), OpenCVColor.colors(image3)])
+    assert_equal [2,2,2], colors.map(&:size)
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: opencv-color
 version: !ruby/object:Gem::Version
-  version: 1.0.2
+  version: 1.0.3
 platform: ruby
 authors:
 - Xiao Li
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-11-30 00:00:00.000000000 Z
+date: 2015-01-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-opencv
@@ -38,6 +38,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.4.0
+- !ruby/object:Gem::Dependency
+  name: terminal-table
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.4.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.4.5
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -83,6 +97,14 @@ files:
 - Rakefile
 - TODO
 - bin/opencv-color
+- color-samples.tb
+- color-samples/blue-light/light-blue1.png
+- color-samples/blue-light/light-blue2.png
+- color-samples/blue-light/light-blue3.png
+- color-samples/blue-light/light-blue4.png
+- color-samples/blue-light/light-blue5.png
+- color-samples/blue-light/light-blue61.png
+- color-samples/blue-light/light-blue7.png
 - color-samples/blue/blue01.png
 - color-samples/blue/blue02.png
 - color-samples/blue/blue2.png
@@ -92,31 +114,14 @@ files:
 - color-samples/blue/blue6.png
 - color-samples/blue/blue7.png
 - color-samples/blue/blue8.png
-- color-samples/dark-yellow/dark-yellow1.png
-- color-samples/dark-yellow/dark-yellow2.png
-- color-samples/dark-yellow/dark-yellow3.png
-- color-samples/dark-yellow/dark-yellow4.png
-- color-samples/dark-yellow/dark-yellow5.png
-- color-samples/dark-yellow/dark-yellow6.png
-- color-samples/dark-yellow/dark-yellow7.png
-- color-samples/dark-yellow/dark-yellow8.png
-- color-samples/dark-yellow/dark-yellow9.png
+- color-samples/green-dark/darkgreen1.png
+- color-samples/green-dark/darkgreen2.png
+- color-samples/green-dark/darkgreen3.png
 - color-samples/green/green1.png
 - color-samples/green/green2.png
 - color-samples/green/green3.png
 - color-samples/green/green4.png
 - color-samples/green/green5.png
-- color-samples/light-blue/light-blue1.png
-- color-samples/light-blue/light-blue2.png
-- color-samples/light-blue/light-blue3.png
-- color-samples/light-blue/light-blue4.png
-- color-samples/light-blue/light-blue5.png
-- color-samples/light-blue/light-blue61.png
-- color-samples/light-blue/light-blue7.png
-- color-samples/light-yellow/light-yellow1.png
-- color-samples/light-yellow/light-yellow2.png
-- color-samples/light-yellow/light-yellow3.png
-- color-samples/light-yellow/light-yellow4.png
 - color-samples/pink/pink1.png
 - color-samples/pink/pink2.png
 - color-samples/pink/pink3.png
@@ -125,9 +130,32 @@ files:
 - color-samples/pink/pink71.png
 - color-samples/pink/pink8.png
 - color-samples/pink/pink9.png
+- color-samples/purple/purple1.png
+- color-samples/purple/purple10.png
+- color-samples/purple/purple2.png
+- color-samples/purple/purple3.png
+- color-samples/purple/purple4.png
+- color-samples/purple/purple5.png
+- color-samples/purple/purple6.png
+- color-samples/purple/purple7.png
+- color-samples/purple/purple8.png
+- color-samples/purple/purple9.png
+- color-samples/yellow-dark/dark-yellow1.png
+- color-samples/yellow-dark/dark-yellow2.png
+- color-samples/yellow-dark/dark-yellow3.png
+- color-samples/yellow-dark/dark-yellow4.png
+- color-samples/yellow-dark/dark-yellow5.png
+- color-samples/yellow-dark/dark-yellow6.png
+- color-samples/yellow-dark/dark-yellow7.png
+- color-samples/yellow-dark/dark-yellow8.png
+- color-samples/yellow-dark/dark-yellow9.png
 - color-samples/yellow-green/yellow-green1.png
 - color-samples/yellow-green/yellow-green2.png
 - color-samples/yellow-green/yellow-green3.png
+- color-samples/yellow-light/light-yellow1.png
+- color-samples/yellow-light/light-yellow2.png
+- color-samples/yellow-light/light-yellow3.png
+- color-samples/yellow-light/light-yellow4.png
 - color-samples/yellow/yellow01.png
 - color-samples/yellow/yellow10.png
 - color-samples/yellow/yellow2.png
@@ -136,6 +164,7 @@ files:
 - lib/opencv-color/objc.erb
 - lib/opencv-color/version.rb
 - opencv-color.gemspec
+- test/opencv_color_test.rb
 homepage: https://github.com/xli/opencv-color
 licenses:
 - MIT
@@ -156,8 +185,9 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.0
+rubygems_version: 2.2.2
 signing_key: 
 specification_version: 4
 summary: Learn OpenCV HSV color range from sample images.
-test_files: []
+test_files:
+- test/opencv_color_test.rb