in tensorflow-inference-docker-lambda/app/app.py [0:0]
def handler(event, context):
r = requests.get(event['url'])
img = tf.image.decode_jpeg(r.content, channels=3)
# Executing inference.
converted_img = tf.image.convert_image_dtype(img, tf.float32)[tf.newaxis, ...]
start_time = time.time()
result = detector(converted_img)
end_time = time.time()
obj = {
'detection_boxes' : result['detection_boxes'].numpy().tolist(),
'detection_scores': result['detection_scores'].numpy().tolist(),
'detection_class_entities': [el.decode('UTF-8') for el in result['detection_class_entities'].numpy()]
}
return {
'statusCode': 200,
'body': json.dumps(obj)
}