Commit inicial

README.md

@@ -0,0 +1,14 @@
+title: Autocrop Image
+emoji: 🐨
+colorFrom: indigo
+colorTo: green
+sdk: gradio
+sdk_version: 3.44.4
+app_file: app.py
+pinned: false
+
+```
+sudo apt-get install python3-opencv
+pip install -r requirements.txt
+python3 main.py
+```

app.py

@@ -0,0 +1,38 @@
+import os
+import gradio as gr
+from services.aws_service import AwsService
+from dotenv import load_dotenv
+from image_crop import crop_faces
+import cv2
+import numpy as np
+
+load_dotenv()
+
+def crop_photoshoot_images(photo_shoot_id):
+  folder = "PhotoShoots/" + str(photo_shoot_id) + "/Inputs"
+  files = AwsService.get_files_from_s3(os.environ.get('AWS_S3_BUCKET'), folder)
+
+  for file in files:
+    s3_object = AwsService.get_image_from_s3(os.environ.get('AWS_S3_BUCKET'), file['Key'])
+    image = s3_object["pil"]
+    key = s3_object["key"]
+    cv2_img = np.array(image)
+    cv2_img = cv2_img[:, :, ::-1].copy()
+    faces = crop_faces(cv2_img)
+    i = 1
+    for face in faces:
+      filename = f'{key}-{i}.jpg'
+      cv2.imwrite(filename, face)
+      AwsService.send_image_to_s3(filename, os.environ.get('AWS_S3_BUCKET'), f'PhotoShoots/{photo_shoot_id}/Croppeds/{key}-{i}.jpg')
+      os.remove(filename)
+      i += 1
+
+  return "done"
+
+iface = gr.Interface(
+  fn=crop_photoshoot_images, 
+  inputs=[gr.Textbox(lines=1, placeholder="Photo Shoot ID")], 
+  outputs=["text"]
+)
+
+iface.launch()

image_crop.py

@@ -0,0 +1,59 @@
+import cv2
+
+def discover_square_crop_points(points, image, step=3, maximum_step=30):
+
+  x = points["x"]
+  y = points["y"]
+  w = points["w"]
+  h = points["h"]
+
+  if step >= maximum_step:
+    return {"initial_x": x, "initial_y": y, "final_x": x + w, "final_y": x + h}
+
+  img_height = image.shape[0]
+  img_width = image.shape[1]
+
+  crop_offset = ((w + h) / 2) / step
+
+  initial_x = x - crop_offset
+  initial_y = y - crop_offset
+  final_x = x + w + crop_offset
+  final_y = y + h + crop_offset
+
+  if initial_x < 0 or initial_y < 0 or final_x > img_width or final_y > img_height:
+    return discover_square_crop_points(points, image, step+1, maximum_step)
+  
+  initial_x = int(initial_x)
+  initial_y = int(initial_y)
+  final_x = int(final_x)
+  final_y = int(final_y)
+
+  print(f"step: {step}")
+  return {"initial_x": initial_x, "initial_y": initial_y, "final_x": final_x, "final_y": final_y}
+
+def crop_faces(cv2_image):
+  gray = cv2.cvtColor(cv2_image, cv2.COLOR_BGR2GRAY)
+  face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_alt2.xml')
+
+  multiScaleValues = [1.01]
+
+  for value in multiScaleValues:
+    faces = face_cascade.detectMultiScale(image=gray, scaleFactor=value, minNeighbors=4, minSize=(512, 512))
+    if len(faces) > 0:
+      break
+
+  cropped_faces = []
+
+  for (x, y, w, h) in faces:
+    points = {"x": x, "y": y, "w": w, "h": h}
+    points = discover_square_crop_points(points, cv2_image)
+
+    initial_x = points["initial_x"]
+    initial_y = points["initial_y"]
+    final_x = points["final_x"]
+    final_y = points["final_y"]
+
+    face = cv2_image[initial_y:final_y, initial_x:final_x]
+    cropped_faces.append(face)
+
+  return cropped_faces

requirements.txt

@@ -0,0 +1,6 @@
+gradio==3.44.4
+markupsafe==2.0.1
+boto3==1.26.25
+botocore==1.29.25
+opencv-python==4.8.1.78
+python-dotenv==1.0.0

services/aws_service.py

@@ -0,0 +1,28 @@
+import os
+import boto3
+from PIL import Image
+from io import BytesIO
+
+class AwsService:
+  def session():
+    return boto3.Session(
+      aws_access_key_id = os.environ.get('AWS_ACCESS_KEY_ID'),
+      aws_secret_access_key = os.environ.get('AWS_SECRET_ACCESS_KEY'),
+      region_name=os.environ.get('AWS_REGION')
+    )
+  
+  def s3_client():
+    return AwsService.session().client('s3')     
+
+  def get_files_from_s3(bucket, prefix):
+    return AwsService.s3_client().list_objects(Bucket=bucket, Prefix=prefix)['Contents'] 
+  
+  def get_image_from_s3(bucket, key):
+    file_byte_string = AwsService.s3_client().get_object(Bucket=bucket, Key=key)['Body'].read()
+    return {
+      'key': key.split('/')[-1].split('.')[0],
+      'pil': Image.open(BytesIO(file_byte_string))
+    }
+  
+  def send_image_to_s3(file, bucket, key):
+    return AwsService.s3_client().upload_file(file, bucket, key, ExtraArgs={'ACL':'public-read', 'ContentType': 'image/png'})