You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
766 B
Python
28 lines
766 B
Python
import json
|
|
import base64
|
|
from PIL import Image
|
|
import io
|
|
from model_handler import ModelHandler
|
|
|
|
def init_context(context):
|
|
context.logger.info("Init context... 0%")
|
|
|
|
# Read the DL model
|
|
model = ModelHandler()
|
|
setattr(context.user_data, 'model', model)
|
|
|
|
context.logger.info("Init context...100%")
|
|
|
|
def handler(context, event):
|
|
context.logger.info("Run SiamMask model")
|
|
data = event.body
|
|
buf = io.BytesIO(base64.b64decode(data["image"].encode('utf-8')))
|
|
shape = data.get("shape")
|
|
state = data.get("state")
|
|
image = Image.open(buf)
|
|
|
|
results = context.user_data.model.infer(image, shape, state)
|
|
|
|
return context.Response(body=json.dumps(results), headers={},
|
|
content_type='application/json', status_code=200)
|