提交 d963be5d 编写于 作者: M MRXLT

fix conflict

...@@ -63,12 +63,16 @@ class WebService(object): ...@@ -63,12 +63,16 @@ class WebService(object):
abort(400) abort(400)
if "fetch" not in request.json: if "fetch" not in request.json:
abort(400) abort(400)
feed, fetch = self.preprocess(request.json, request.json["fetch"]) try:
feed, fetch = self.preprocess(request.json,
request.json["fetch"])
if isinstance(feed, list): if isinstance(feed, list):
fetch_map_batch = client_service.batch_predict( fetch_map_batch = client_service.predict(
feed_batch=feed, fetch=fetch) feed_batch=feed, fetch=fetch)
fetch_map_batch = self.postprocess( fetch_map_batch = self.postprocess(
feed=request.json, fetch=fetch, fetch_map=fetch_map_batch) feed=request.json,
fetch=fetch,
fetch_map=fetch_map_batch)
result = {"result": fetch_map_batch} result = {"result": fetch_map_batch}
elif isinstance(feed, dict): elif isinstance(feed, dict):
if "fetch" in feed: if "fetch" in feed:
...@@ -76,6 +80,8 @@ class WebService(object): ...@@ -76,6 +80,8 @@ class WebService(object):
fetch_map = client_service.predict(feed=feed, fetch=fetch) fetch_map = client_service.predict(feed=feed, fetch=fetch)
result = self.postprocess( result = self.postprocess(
feed=request.json, fetch=fetch, fetch_map=fetch_map) feed=request.json, fetch=fetch, fetch_map=fetch_map)
except ValueError:
result = {"result": "Request Value Error"}
return result return result
app_instance.run(host="0.0.0.0", app_instance.run(host="0.0.0.0",
......
...@@ -94,12 +94,16 @@ class WebService(object): ...@@ -94,12 +94,16 @@ class WebService(object):
client.connect([endpoint]) client.connect([endpoint])
while True: while True:
request_json = inputqueue.get() request_json = inputqueue.get()
feed, fetch = self.preprocess(request_json, request_json["fetch"]) try:
feed, fetch = self.preprocess(request_json,
request_json["fetch"])
if isinstance(feed, list): if isinstance(feed, list):
fetch_map_batch = client.batch_predict( fetch_map_batch = client.predict(
feed_batch=feed, fetch=fetch) feed_batch=feed, fetch=fetch)
fetch_map_batch = self.postprocess( fetch_map_batch = self.postprocess(
feed=request_json, fetch=fetch, fetch_map=fetch_map_batch) feed=request_json,
fetch=fetch,
fetch_map=fetch_map_batch)
result = {"result": fetch_map_batch} result = {"result": fetch_map_batch}
elif isinstance(feed, dict): elif isinstance(feed, dict):
if "fetch" in feed: if "fetch" in feed:
...@@ -107,8 +111,9 @@ class WebService(object): ...@@ -107,8 +111,9 @@ class WebService(object):
fetch_map = client.predict(feed=feed, fetch=fetch) fetch_map = client.predict(feed=feed, fetch=fetch)
result = self.postprocess( result = self.postprocess(
feed=request_json, fetch=fetch, fetch_map=fetch_map) feed=request_json, fetch=fetch, fetch_map=fetch_map)
self.output_queue.put(result) self.output_queue.put(result)
except ValueError:
self.output_queue.put(-1)
def _launch_web_service(self, gpu_num): def _launch_web_service(self, gpu_num):
app_instance = Flask(__name__) app_instance = Flask(__name__)
...@@ -152,6 +157,8 @@ class WebService(object): ...@@ -152,6 +157,8 @@ class WebService(object):
if self.idx >= len(self.gpus): if self.idx >= len(self.gpus):
self.idx = 0 self.idx = 0
result = self.output_queue.get() result = self.output_queue.get()
if not isinstance(result, dict) and result == -1:
result = {"result": "Request Value Error"}
return result return result
''' '''
feed, fetch = self.preprocess(request.json, request.json["fetch"]) feed, fetch = self.preprocess(request.json, request.json["fetch"])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册