提交 46641c63 编写于 作者: G guosheng

Fix infer when input is empty in v2/inference.py

上级 e82fda25
......@@ -96,6 +96,9 @@ class Inference(object):
for i, item in enumerate(result):
retv[i].append(item)
if retv == None:
return []
if flatten_result:
retv = [numpy.concatenate(out) for out in retv]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册