未验证 提交 d8efe87c 编写于 作者: M MRXLT 提交者: GitHub

Merge branch 'develop' into cuda-10

...@@ -65,7 +65,7 @@ image_service = ImageService(name="image") ...@@ -65,7 +65,7 @@ image_service = ImageService(name="image")
image_service.load_model_config(sys.argv[1]) image_service.load_model_config(sys.argv[1])
image_service.init_imagenet_setting() image_service.init_imagenet_setting()
if device == "gpu": if device == "gpu":
image_service.set_gpus("0,1") image_service.set_gpus("0")
image_service.prepare_server( image_service.prepare_server(
workdir="workdir", port=int(sys.argv[3]), device=device) workdir="workdir", port=int(sys.argv[3]), device=device)
image_service.run_rpc_service() image_service.run_rpc_service()
......
...@@ -204,4 +204,6 @@ class WebService(object): ...@@ -204,4 +204,6 @@ class WebService(object):
return feed, fetch return feed, fetch
def postprocess(self, feed=[], fetch=[], fetch_map=None): def postprocess(self, feed=[], fetch=[], fetch_map=None):
for key in fetch_map.iterkeys():
fetch_map[key] = fetch_map[key].tolist()
return fetch_map return fetch_map
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册