spyglass 0.0.3 → 0.0.4

data/ext/spyglass/bgslib_prati_mediod_bgs.h

@@ -0,0 +1,142 @@
+/*
+This file is part of BGSLibrary.
+
+BGSLibrary is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+BGSLibrary is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with BGSLibrary.  If not, see <http://www.gnu.org/licenses/>.
+*/
+/****************************************************************************
+*
+* PratiMediodBGS.hpp
+*
+* Purpose: Implementation of the temporal median background 
+*		  		 subtraction algorithm described in:
+*
+* [1] "Detecting Moving Objects, Shosts, and Shadows in Video Stream"
+* 			by R. Cucchiara et al (2003)
+*
+* [2] "Reliable Background Suppression for Complex Scenes"
+*				by S. Calderara et al (2006)
+*
+* Author: Donovan Parks, September 2007
+*
+* Please note that this is not an implementation of the complete system 
+* given in the above papers. It simply implements the temporal media background
+* subtraction algorithm.
+
+Example:
+Algorithms::BackgroundSubtraction::PratiParams params;
+params.SetFrameSize(width, height);
+params.LowThreshold() = 30;
+params.HighThreshold() = 2*params.LowThreshold();	// Note: high threshold is used by post-processing 
+params.SamplingRate() = 5;
+params.HistorySize() = 16;
+params.Weight() = 5;
+
+Algorithms::BackgroundSubtraction::PratiMediodBGS bgs;
+bgs.Initalize(params);
+******************************************************************************/
+
+#ifndef PRATI_MEDIA_BGS_H
+#define PRATI_MEDIA_BGS_H
+
+#include <vector>
+#include "bgslib_bgs.h"
+
+namespace Algorithms
+{
+  namespace BackgroundSubtraction
+  {
+    // --- Parameters used by the Prati Mediod BGS algorithm ---
+    class PratiParams : public BgsParams
+    {
+    public:
+      unsigned int &LowThreshold() { return m_low_threshold; }
+      unsigned int &HighThreshold() { return m_high_threshold; }
+
+      int &Weight() { return m_weight; }
+      int &SamplingRate() { return m_sampling_rate; }
+      int &HistorySize() { return m_history_size; }
+
+    private:
+      // The low threshold is used to supress noise. The high thresohld is used 
+      // to find pixels highly likely to be foreground. This implementation uses an L-inf 
+      // distance measure and a pixel p is considered F/G if D(I(p), B(p)) > threshold. 
+      // The two threshold maps are combined as in [2].
+      unsigned int m_low_threshold;
+      unsigned int m_high_threshold;
+
+      // The weight parameter controls the amount of influence given to previous background samples
+      // see w_b in equation (2) of [1]
+      // in [2] this value is set to 1
+      int m_weight;
+
+      // Number of samples to consider when calculating temporal mediod value
+      int m_history_size;
+
+      // Rate at which to obtain new samples
+      int m_sampling_rate;
+    };
+
+    // --- Prati Mediod BGS algorithm ---
+    class PratiMediodBGS : public Bgs 
+    {
+    private:	
+      // sum of L-inf distances from a sample point to all other sample points
+      struct MEDIAN_BUFFER
+      {
+        std::vector<RgbPixel> pixels;		// vector of pixels at give location in image
+        std::vector<int> dist;					// distance from pixel to all other pixels
+        int pos;												// current position in circular buffer
+
+        RgbPixel median;								// median at this pixel location
+        int medianDist;									// distance from median pixel to all other pixels
+      };
+
+    public:
+      PratiMediodBGS();
+      ~PratiMediodBGS();
+
+      void Initalize(const BgsParams& param);
+
+      void InitModel(const RgbImage& data);
+      void Subtract(int frame_num, const RgbImage& data,  
+        BwImage& low_threshold_mask, BwImage& high_threshold_mask);	
+      void Update(int frame_num, const RgbImage& data,  const BwImage& update_mask);
+
+      RgbImage* Background() { return &m_background; }
+
+    private:	
+      MEDIAN_BUFFER* m_median_buffer;
+
+      void CalculateMasks(int r, int c, const RgbPixel& pixel);
+      void Combine(const BwImage& low_mask, const BwImage& high_mask, BwImage& output);
+      void UpdateMediod(int r, int c, const RgbImage& new_frame, int& dist);
+
+      PratiParams m_params;
+
+      RgbImage m_background;
+
+      BwImage m_mask_low_threshold;
+      BwImage m_mask_high_threshold;
+    };
+
+  };
+};
+
+#endif
+
+
+
+
+
+

data/ext/spyglass/contour.cc

@@ -66,8 +66,6 @@ namespace Spyglass {
       std::vector<cv::Point> contour = to_value_vector(SG_GET_CONTOUR(self));
       cv::approxPolyDP(contour, contour, cv::arcLength(cv::Mat(contour), true) * 0.02, true);
       if (contour.size() != 4) {
-        // TODO: Throw exception here?
-        std::cout << "The object is not quadrilateral!" << std::endl;
         return Qnil;
       }
 

data/ext/spyglass/extconf.rb

@@ -4,3 +4,4 @@ $CFLAGS     = ENV["CFLAGS"].to_s  + " " + `pkg-config opencv --cflags`.chomp
 $LOCAL_LIBS = ENV["LIBS"].to_s    + " " + `pkg-config opencv --libs`.chomp
 
 create_makefile("spyglass/spyglass")
+

data/ext/spyglass/prelude.h

@@ -15,7 +15,7 @@
     return ptr; \
   }
 
-SG_GEN_GET_OBJECT_FUNCTION(SG_GET_BG_SUBTRACTOR,  cv::BackgroundSubtractorMOG2);
+SG_GEN_GET_OBJECT_FUNCTION(SG_GET_BG_SUBTRACTOR,  cv::BackgroundSubtractor);
 SG_GEN_GET_OBJECT_FUNCTION(SG_GET_CLASSIFIER,     cv::CascadeClassifier);
 SG_GEN_GET_OBJECT_FUNCTION(SG_GET_COLOR,          cv::Scalar);
 SG_GEN_GET_OBJECT_FUNCTION(SG_GET_CONTOUR,        std::vector<cv::Point *>);

data/ext/spyglass/spyglass.cc

@@ -19,7 +19,7 @@ extern "C" {
     Spyglass::define_ruby_module();
     Spyglass::ColorSpace::define_ruby_module();
 
-    Spyglass::BackgroundSubtractor::define_ruby_class();
+    Spyglass::BackgroundSubtractor::define_ruby_types();
     Spyglass::CascadeClassifier::define_ruby_class();
     Spyglass::Color::define_ruby_class();
     Spyglass::Contour::define_ruby_class();

data/lib/spyglass/version.rb

@@ -1,3 +1,3 @@
-module Spyglass 
-  VERSION = "0.0.3"
+module Spyglass
+  VERSION = "0.0.4"
 end

data/spec/spyglass/background_subtractor_spec.rb

@@ -1,23 +1,28 @@
 require 'spec_helper'
 
-describe Spyglass::BackgroundSubtractor do
-  let(:bg) { Spyglass::BackgroundSubtractor.new }
-  let(:lena) { Spyglass::Image.load(fixture_path('lena.jpg')) }
+[Spyglass::BackgroundSubtractor::MOG,
+ Spyglass::BackgroundSubtractor::MOG2,
+ Spyglass::BackgroundSubtractor::GMG,
+ Spyglass::BackgroundSubtractor::PratiMediod].each do |algorithm|
+   describe algorithm do
+     let(:bg) { described_class.new }
+     let(:lena) { Spyglass::Image.load(fixture_path('lena.jpg')) }
 
-  describe '.new' do
-    it 'should work with no arguments' do
-      expect( bg ).to be_a Spyglass::BackgroundSubtractor
-    end
+     describe '.new' do
+       it 'should work with no arguments' do
+         expect( bg ).to be_a described_class
+       end
 
-    it 'should work with an options hash' do
-      expect { Spyglass::BackgroundSubtractor.new(history: 50, threshold: 64) }.not_to raise_error
-    end
-  end
+       it 'should work with an options hash' do
+         expect { algorithm.new(history: 50, threshold: 64) }.not_to raise_error
+       end
+     end
 
-  describe '#subtract' do
-    it 'should return a Spyglass::Image' do
-      delta = bg.subtract(lena, 1)
-      expect( delta ).to be_a Spyglass::Image
-    end
-  end
-end
+     describe '#subtract' do
+       it 'should return a Spyglass::Image' do
+         delta = bg.subtract(lena, 1)
+         expect( delta ).to be_a Spyglass::Image
+       end
+     end
+   end
+ end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: spyglass
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - André Medeiros
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-09-27 00:00:00.000000000 Z
+date: 2013-10-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -91,6 +91,14 @@ files:
 - examples/video_capture.rb
 - ext/spyglass/background_subtractor.cc
 - ext/spyglass/background_subtractor.h
+- ext/spyglass/bgslib_bgs.h
+- ext/spyglass/bgslib_bgs_params.h
+- ext/spyglass/bgslib_dp_prati_mediod_bgs.cc
+- ext/spyglass/bgslib_dp_prati_mediod_bgs.h
+- ext/spyglass/bgslib_image.cc
+- ext/spyglass/bgslib_image.h
+- ext/spyglass/bgslib_prati_mediod_bgs.cc
+- ext/spyglass/bgslib_prati_mediod_bgs.h
 - ext/spyglass/cascade_classifier.cc
 - ext/spyglass/cascade_classifier.h
 - ext/spyglass/color.cc