paperclip-facecrop 0.0.1 → 0.0.2

data/README.md

@@ -13,19 +13,47 @@ Requirements:
 
 Installation:
 -------------
-- Install the gem
-          
-          `gem install paperclip-facecrop'`
-          
-- Add to `Gemfile`
+- Add to your application `Gemfile`
 
           gem 'paperclip-facecrop'
+          
+- Type
+          
+          bundle install
 
 - Write an initializer setting the path of the haarcascade filters(`initializers/paperclip.rb` for example):   
 
-          Paperclip::FaceCrop.classifiers = ["/usr/local/share/opencv/haarcascades/haarcascade_frontalface_alt_tree.xml"]
+          Paperclip::FaceCrop.classifiers = {
+            :face => ["/usr/local/share/opencv/haarcascades/haarcascade_frontalface_alt.xml"]
+          }
+              
+    You can use more than one filter to try more accurate searches:
+    
+          Paperclip::FaceCrop.classifiers = {
+            :face => [
+              "/usr/local/share/opencv/haarcascades/haarcascade_frontalface_alt.xml",
+              "/usr/local/share/opencv/haarcascades/haarcascade_frontalface_alt_tree.xml",
+              "/usr/local/share/opencv/haarcascades/haarcascade_profileface.xml"
+            ]
+          }
+
+    In order to try to avoid some false positives, you can also specify other classifiers to detect other parts of the face. In that case, 
+    only the found areas that contain parts like a mouth, an eye or a nose will be considered a face:
+    
+          Paperclip::FaceCrop.classifiers = {
+            :face => [
+              "/usr/local/share/opencv/haarcascades/haarcascade_frontalface_alt.xml",
+              "/usr/local/share/opencv/haarcascades/haarcascade_frontalface_alt_tree.xml",
+              "/usr/local/share/opencv/haarcascades/haarcascade_profileface.xml"
+            ],
+            :parts => [
+              "/usr/local/share/opencv/haarcascades/haarcascade_mcs_nose.xml",
+              "/usr/local/share/opencv/haarcascades/haarcascade_mcs_lefteye.xml",
+              "/usr/local/share/opencv/haarcascades/haarcascade_mcs_righteye.xml"
+            ]
+          }
+    
     
-You can use more than one filter if you want in order to try more accurate searches.
 
 Usage:
 ------
@@ -40,6 +68,13 @@ Just specify your image styles as usual and set :face_crop as the processor:
     
 In case no faces were found, it will behave simply as the `Paperclip::Thumbnail` processor
 
+
+You can also set the debug mode to draw on the image the detected regions:
+    
+    Paperclip::FaceCrop.debug = (Rails.env == 'development')
+    
+![](https://github.com/dagi3d/paperclip-facecrop/raw/master/README_example_b.jpg)
+
 Credits:
 --------
 Copyright (c) 2011 Borja Martín Sánchez de Vivar <borjamREMOVETHIS@dagi3d.net> - <http://dagi3d.net>, released under the MIT license  

data/lib/paperclip-facecrop.rb

@@ -1,33 +1,57 @@
 require 'opencv'
 
+class OpenCV::CvAvgComp
+  def to_s
+    "#{self.x},#{self.y}-#{self.width}x#{self.height}"
+  end
+end
+
 class Paperclip::FaceCrop < Paperclip::Thumbnail
+
+  @@debug = false
   
   cattr_accessor :classifiers
-
+  cattr_accessor :debug
+  
   def initialize(file, options = {}, attachment = nil)
     super(file, options, attachment)
     
-    x_coords = []
-    y_coords = []
-    widths   = []
-    heights  = []
+    faces_regions = []
+    faces_parts_regions = []
+        
+    raise "No classifiers were defined" if self.classifiers.nil?
     
-    raise "No classifiers were defined" if Paperclip::FaceCrop.classifiers.nil?
+    image = OpenCV::IplImage.load(file.path, 1)
     
-    image = OpenCV::IplImage.load(file.path)
+    faces_regions = detect_regions(image, self.classifiers[:face])
     
-    Paperclip::FaceCrop.classifiers.each do |classifier|
-      detector = OpenCV::CvHaarClassifierCascade::load(classifier)
-      detector.detect_objects(image) do |region|
-        x_coords << region.top_left.x << region.bottom_right.x
-        y_coords << region.top_left.y << region.bottom_right.y
-        widths << region.width
-        heights << region.height
+    #Paperclip::FaceCrop.classifiers[:nose]
+    unless self.classifiers[:parts].nil?
+      faces_parts_regions = detect_regions(image, self.classifiers[:parts], OpenCV::CvColor::Red)
+      
+      faces_regions.reject! do |face_region|
+        region = faces_parts_regions.detect do |part_region|
+          regions_overlap?(face_region, part_region)
+        end
+        
+        region.nil?
       end
     end
     
-    @has_faces = x_coords.size > 0
+    x_coords = []
+    y_coords = []
+    widths   = []
+    heights  = []
+    
+    faces_regions.each do |region|
+      x_coords << region.top_left.x << region.bottom_right.x
+      y_coords << region.top_left.y << region.bottom_right.y
+      widths << region.width
+      heights << region.height
+    end
     
+    @has_faces = faces_regions.size > 0
+     
     if @has_faces
       @top_left_x = x_coords.min
       @top_left_y = y_coords.min
@@ -37,7 +61,7 @@ class Paperclip::FaceCrop < Paperclip::Thumbnail
       # average faces areas
       average_face_width  = widths.sum / widths.size
       average_face_height = heights.sum / heights.size
-    
+      
       # calculating the surrounding margin of the area that covers all the found faces
       #
       
@@ -53,25 +77,45 @@ class Paperclip::FaceCrop < Paperclip::Thumbnail
       
       # if the new area is smaller than the target geometry, it's scaled so the final image isn't resampled
       #
-      if @faces_width < @target_geometry.width
+      if @faces_width < @target_geometry.width 
         delta_width = (@target_geometry.width - @faces_width) / 2
         @top_left_x -= delta_width
         @bottom_right_x += delta_width
         calculate_bounds
       end
       
-      if @faces_height < @target_geometry.height
+      #raise (@target_geometry.height > 0 and @faces_height < @target_geometry.height).to_s
+      
+      if (@target_geometry.height > 0 and @faces_height < @target_geometry.height)
         delta_height = (@target_geometry.height - @faces_height) / 2
         @top_left_y -= delta_height
         @bottom_right_y += delta_height
         calculate_bounds
       end
       
+      @faces_height = @faces_width if @target_geometry.height == 0
+      
       @current_geometry = Paperclip::Geometry.new(@faces_width, @faces_height)
     end
     
   end
-
+  
+  
+  def transformation_command
+    return super unless @has_faces
+    
+    scale, crop = @current_geometry.transformation_to(@target_geometry, crop?)
+    faces_crop = "%dx%d+%d+%d" % [@faces_width, @faces_height, @top_left_x, @top_left_y]
+    
+    trans = []
+    trans << "-crop" << %["#{faces_crop}"] << "+repage"
+    trans << "-resize" << %["#{scale}"] unless scale.nil? || scale.empty?
+    trans << "-crop" << %["#{crop}"] << "+repage" if crop
+    trans
+  end
+  
+  private
+  
   # calculate_bounds
   #
   def calculate_bounds
@@ -85,17 +129,30 @@ class Paperclip::FaceCrop < Paperclip::Thumbnail
     @faces_height = @bottom_right_y - @top_left_y
   end
   
-  
-  def transformation_command
-    return super unless @has_faces
+  # detect_regions
+  #
+  def detect_regions(image, classifiers, color = OpenCV::CvColor::Blue)
+    regions = []
     
-    scale, crop = @current_geometry.transformation_to(@target_geometry, crop?)
-    faces_crop = "%dx%d+%d+%d" % [@faces_width, @faces_height, @top_left_x, @top_left_y]
+    classifiers.each do |classifier|
+      detector = OpenCV::CvHaarClassifierCascade::load(classifier)
+      detector.detect_objects(image) do |region| 
+        regions << region 
+        image.rectangle!(region.top_left, region.bottom_right, :color => color) if self.debug 
+      end
+    end
     
-    trans = []
-    trans << "-crop" << %["#{faces_crop}"] << "+repage"
-    trans << "-resize" << %["#{scale}"] unless scale.nil? || scale.empty?
-    trans << "-crop" << %["#{crop}"] << "+repage" if crop
-    trans
+    if self.debug 
+      image.save_image(@file.path)
+      Rails.logger.info(regions)
+    end
+    
+    regions
+  end
+  
+  def regions_overlap?(r1, r2)
+    dx = (r2.x - r1.x).abs + r2.width
+    dy = (r2.y - r1.y).abs + r2.height;
+    return (dx < r1.width + r2.width && dy < r1.height + r2.height);
   end
 end

data/lib/version.rb

@@ -1,5 +1,5 @@
 class Paperclip
   class FaceCrop
-    VERSION = "0.0.1"
+    VERSION = "0.0.2"
   end
 end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 1
-  version: 0.0.1
+  - 2
+  version: 0.0.2
 platform: ruby
 authors: 
 - "Borja Mart\xC3\xADn"
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-05-03 00:00:00 +02:00
+date: 2011-05-06 00:00:00 +02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -58,6 +58,7 @@ files:
 - Gemfile
 - README.md
 - README_example.jpg
+- README_example_b.jpg
 - Rakefile
 - lib/paperclip-facecrop.rb
 - lib/version.rb