image-process-opencv-lambda

srcecde · srcecde · commit db21322ee5ef · 2019-04-06T10:54:45.000+05:30
diff --git a/lambda/lambda_image_process_opencv.py b/lambda/lambda_image_process_opencv.py
@@ -0,0 +1,28 @@
+"""
+-*- coding: utf-8 -*-
+========================
+AWS Lambda
+========================
+Contributor: Chirag Rathod (Srce Cde)
+========================
+"""
+import cv2
+import numpy as np
+import boto3
+
+
+def lambda_handler(event, context):
+    s3 = boto3.client("s3")
+    bucket_name = "bucket_name"
+    file_obj = s3.get_object(Bucket=bucket_name, Key="obj_scene.jpeg")
+    file_content = file_obj["Body"].read()
+
+    np_array = np.fromstring(file_content, np.uint8)
+    image_np = cv2.imdecode(np_array, cv2.IMREAD_COLOR)
+
+    gray = cv2.cvtColor(image_np, cv2.COLOR_BGR2GRAY)
+    cv2.imwrite("/tmp/gray_obj.jpg", gray)
+
+    s3.put_object(Bucket=bucket_name, Key="grayscale.jpg", Body=open("/tmp/gray_obj.jpg", "rb").read())
+
+    return "Thanks"
