face_cropper 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 53597329a198f8e0bed4fd9c64cba57ff000acde
-  data.tar.gz: 0f4d4a56e77c6823dd755c2575ca3cdac164c396
+  metadata.gz: 702e3d2fbe93f586b47c7e3f2848f05729cc81b5
+  data.tar.gz: c5b462c7ca47f0d8b535a9cb16074f62de01cbb9
 SHA512:
-  metadata.gz: cd10291b7478597d330694b2f0a769ab8d31ed44a74d8615fd3cda76731775946a5e451d0c93c9830594f2660fed53ea3d5c9f239d50e6e624711a85570230b4
-  data.tar.gz: 0fda9a49cc182c72d4732c84d799190af447854bacfeb2e8ff6c16525fb9b707d8db928d5c1519c64ef0cd767c794f453071a7fd43bd8989efc9893e35245a09
+  metadata.gz: 8d730b640816b8195ee5ccf5e558838cefc9462eef2e2dc9ef6f1e1aa7aa237cdaec11d201bf77069986c4a43a826ad92f8a31381a887e75560561f2b2d3412f
+  data.tar.gz: 5a1fa3cab16bbefb0294cd8f8aaf141ea5fdb5a4f3498d4b8e5172de3494b2fb8d1ad28a75f0e0b0bf4452d23d17b054162a81592f8d340b5231d5736c230248

data/lib/face_cropper.rb

@@ -1,35 +1,32 @@
 require "face_cropper/version"
+require "face_cropper/aws_rekognition_face_detector"
+require "face_cropper/face_box"
 require 'aws-sdk'
-require 'mini_magick'
-require 'yaml'
 
 class FaceCropper
   def initialize(params)
     @from_bucket = params[:from_bucket]
     @to_bucket   = params[:to_bucket]
     @image_key   = params[:image_key]
+    @face_boxis  = params[:face_details]
   end
 
   def crop_and_upload!
-    faces = detect_faces!
-    puts faces.to_yaml
+    faces = @face_boxis || detect_faces!
+
     tmp_original_image_path = download_original_image!
-    upload_faces!(faces, tmp_original_image_path)
+    crop_faces!(faces, tmp_original_image_path)
   end
 
   private
 
+  def debug_print(faces)
+    pp faces if ENV['API_DEBUG']
+  end
+
   def detect_faces!
-    rekognition = Aws::Rekognition::Client.new(region: 'us-east-1')
-
-    rekognition.detect_faces(
-      image: {
-        s3_object: {
-          bucket: @from_bucket,
-          name:   @image_key
-        }
-      }
-    )
+    detector = AwsRekognitionFaceDetector.new(bucket: @from_bucket, image_key: @image_key)
+    detector.dcetect!.tap {|r| debug_print(r) }
   end
 
   def download_original_image!
@@ -39,20 +36,20 @@ class FaceCropper
     end
   end
 
-  def upload_faces!(faces, image_path)
+  def crop_faces!(faces, image_path)
     faces.face_details.each_with_index do |detail, index|
-      image = MiniMagick::Image.open(image_path)
-
-      w = detail.bounding_box.width  * image.width
-      h = detail.bounding_box.height * image.height
-      x = detail.bounding_box.top    * image.height
-      y = detail.bounding_box.left   * image.width
-      crop_params = "#{w.to_i}x#{h.to_i}+#{y.to_i}+#{x.to_i}"
-
-      image.crop(crop_params)
-      crop_file = "#{index}_#{@image_key}"
-      image.write(crop_file)
-      s3_client.put_object(bucket: @to_bucket, key: crop_file, body: File.read(crop_file))
+      face_box = FaceBox.new(
+        width:  detail.bounding_box.width,
+        height: detail.bounding_box.height,
+        top:    detail.bounding_box.top,
+        left:   detail.bounding_box.left
+      )
+      crop_file = face_box.crop_face!(image_path)
+
+      if @to_bucket
+        s3_client.put_object(bucket: @to_bucket, key: crop_file, body: File.read(crop_file))
+        File.unlink crop_file
+      end
     end
   end
 

data/lib/face_cropper/aws_rekognition_face_detector.rb

@@ -0,0 +1,24 @@
+require 'aws-sdk'
+require 'pp'
+
+class FaceCropper
+  class AwsRekognitionFaceDetector
+    def initialize(bucket:, image_key:)
+      @bucket     = bucket
+      @imaget_key = image_key
+    end
+
+    def dcetect!
+      rekognition = Aws::Rekognition::Client.new(region: 'us-east-1')
+
+      rekognition.detect_faces(
+        image: {
+          s3_object: {
+            bucket: @from_bucket,
+            name:   @image_key
+          }
+        }
+      )
+    end
+  end
+end

data/lib/face_cropper/face_box.rb

@@ -0,0 +1,36 @@
+require 'mini_magick'
+
+class FaceCropper
+  class FaceBox
+    attr_reader :top, :left, :height, :width
+
+    def initialize(top: , left: , height: , width:)
+      @top    = top
+      @left   = left
+      @height = height
+      @width  = width
+    end
+
+    def crop_face!(image_path)
+      image = MiniMagick::Image.open(image_path)
+      position = calculate_position(image_width: image.width, image_height: image.height)
+
+      crop_params = "#{position[:width]}x#{position[:height]}+#{position[:y]}+#{position[:x]}"
+
+      image.crop(crop_params)
+      crop_file = "#{crop_params}_#{@image_key}"
+      image.write(crop_file)
+
+      crop_file
+    end
+
+    def calculate_position(image_width: , image_height:)
+      {
+        width:  (@width  * image_width).to_i,
+        height: (@height * image.height).to_i,
+        x:      (@top    * image.height).to_i,
+        y:      (@left   * image.width).to_i
+      }
+    end
+  end
+end

data/lib/face_cropper/version.rb

@@ -1,3 +1,3 @@
 class FaceCropper
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: face_cropper
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - takkanm
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-02-08 00:00:00.000000000 Z
+date: 2017-02-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk
@@ -112,6 +112,8 @@ files:
 - bin/setup
 - face_cropper.gemspec
 - lib/face_cropper.rb
+- lib/face_cropper/aws_rekognition_face_detector.rb
+- lib/face_cropper/face_box.rb
 - lib/face_cropper/version.rb
 homepage: https://github.com/takkanm/face_cropper
 licenses: