提交 c1b47b20 编写于 作者: T Tao Luo 提交者: GitHub

Merge pull request #1799 from luotao1/infer

fix bug in inference.py
...@@ -49,7 +49,7 @@ class Inference(object): ...@@ -49,7 +49,7 @@ class Inference(object):
retv = None retv = None
for result in self.iter_infer_field(field=field, **kwargs): for result in self.iter_infer_field(field=field, **kwargs):
if retv is None: if retv is None:
retv = [[]] * len(result) retv = [[] for i in xrange(len(result))]
for i, item in enumerate(result): for i, item in enumerate(result):
retv[i].append(item) retv[i].append(item)
retv = [numpy.concatenate(out) for out in retv] 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.
先完成此消息的编辑!
想要评论请 注册