1 Star 0 Fork 1

清欢/cvat

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
main.py 1020 Bytes
一键复制 编辑 原始数据 按行查看 历史
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%")
model = ModelHandler()
setattr(context.user_data, 'model', model)
context.logger.info("Init context...100%")
def handler(context, event):
context.logger.info("Run person-reidentification-retail-0300 model")
data = event.body
buf0 = io.BytesIO(base64.b64decode(data["image0"].encode('utf-8')))
buf1 = io.BytesIO(base64.b64decode(data["image1"].encode('utf-8')))
threshold = float(data.get("threshold", 0.5))
max_distance = float(data.get("max_distance", 50))
image0 = Image.open(buf0)
image1 = Image.open(buf1)
boxes0 = data["boxes0"]
boxes1 = data["boxes1"]
results = context.user_data.model.infer(image0, boxes0,
image1, boxes1, threshold, max_distance)
return context.Response(body=json.dumps(results), headers={},
content_type='application/json', status_code=200)
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/irving512/cvat.git
git@gitee.com:irving512/cvat.git
irving512
cvat
cvat
develop-20201204-aliyun

搜索帮助