def sage_img():
imgFile = request.files["myImage"]
if "myImage" not in request.files:
return jsonify({'error': 'No data provided.'}), 400
nparr = np.frombuffer(imgFile.read(), dtype=np.uint8)
imgInBGR = cv2.imdecode(nparr, flags=1)
imgIn = cv2.cvtColor(imgInBGR, cv2.COLOR_BGR2RGB)
imgOut = cartoonize(imgIn)
im = Image.fromarray(imgOut)
# create file-object in memory
file_object = io.BytesIO()
# write PNG in file-object
im.save(file_object, "PNG")
# move to beginning of file so `send_file()` it will read from start
file_object.seek(0)
# send file directly:
return send_file(file_object,
attachment_filename='processedImg.PNG',
mimetype='image/PNG')
#!/usr/bin/env python
from cartoonizer import cartoonize
import cv2
import os
import time
in_dir = './imgs/input'
out_dir = './imgs/output'
os.mkdir(out_dir)
for f in os.listdir(in_dir):
image = cv2.imread(os.path.join(in_dir, f))
print('==============')
print(f)
start_time = time.time()
output = cartoonize(image)
end_time = time.time()
print("time: {0}s".format(end_time - start_time))
name = os.path.basename(f)
tmp = os.path.splitext(name)
name = tmp[0] + "_cartoon" + tmp[1]
name = os.path.join(out_dir, name)
print("write to {0}".format(name))
cv2.imwrite(name, output)
#!/usr/bin/env python
from cartoonizer import cartoonize
import cv2
import os
import time
in_dir = './imgs/input'
out_dir = './imgs/output'
for f in os.listdir(in_dir):
image = cv2.imread(os.path.join(in_dir, f))
print('==============')
print(f)
start_time = time.time()
output = cartoonize(image)
end_time = time.time()
print("time: {0}s".format(end_time-start_time))
name = os.path.basename(f)
tmp = os.path.splitext(name)
name = tmp[0]+"_cartoon" + tmp[1]
name = os.path.join(out_dir, name)
cv2.imwrite(name, output)
def lambda_handler(event, context):
print("[DEBUG] event = {}".format(event))
src_filename = event.get("name", None)
min_v = event.get("min", 100)
max_v = event.get("max", 200)
change_fullimage = event.get("min", False)
filename_set = os.path.splitext(src_filename)
basename = filename_set[0]
ext = filename_set[1]
h = basename.split("/")[1]
#
# local files
#
down_filename = '/tmp/my_image{}'.format(ext)
conv_filename = '/tmp/normal_cartoon{}'.format(ext)
down_jsonfile = '/tmp/normal_cartoon.json'
#
# S3 files
#
s3_filename = 'public/{}/normal-cartoon{}'.format(h, ext)
s3_paramfile = 'public/{}/normal-cartoon.json'.format(h)
if os.path.exists(down_filename):
os.remove(down_filename)
if os.path.exists(conv_filename):
os.remove(conv_filename)
if os.path.exists(down_jsonfile):
os.remove(down_jsonfile)
#
# Download source image from S3.
#
s3 = boto3.client('s3')
BUCKET_NAME = os.environ.get("BUCKET_NAME")
S3_KEY = src_filename
try:
s3.download_file(BUCKET_NAME, S3_KEY, down_filename)
except botocore.exceptions.ClientError as e:
if e.response['Error']['Code'] == "404":
print("===error message ===> {}".format(e))
print("The object does not exist: s3://{}/{}".format(
BUCKET_NAME, S3_KEY))
else:
raise
image = cv2.imread(down_filename)
output = cartoonizer.cartoonize(image, min_v, max_v)
cv2.imwrite(conv_filename, output)
#
# Upload the converion image to S3
#
s3.upload_file(conv_filename, BUCKET_NAME, s3_filename)
#
# Save params for sketchify whenever converting fullimage.
#
j = {"min": min_v, "max": max_v}
if change_fullimage != False:
with open(down_jsonfile, 'w') as f:
f.write(json.dumps(j))
s3.upload_file(down_jsonfile, BUCKET_NAME, s3_paramfile)
images = {
"source":
S3_URL.format(bucketName=BUCKET_NAME, keyName=src_filename),
"params":
j,
"dest":
DEST_S3_URL.format(bucketName=BUCKET_NAME,
keyName=s3_filename,
timeStamp=time.time())
}
return {"statusCode": 200, "body": {"images": images}}