提交 6e6586f5 编写于 作者: P parap1uie-s 提交者: Walter

Fixed the incorrect infer outputs

上级 57504f26
...@@ -520,10 +520,11 @@ class Engine(object): ...@@ -520,10 +520,11 @@ class Engine(object):
out = out["logits"] out = out["logits"]
if isinstance(out, dict) and "output" in out: if isinstance(out, dict) and "output" in out:
out = out["output"] out = out["output"]
result = self.postprocess_func(out, image_file_list)
print(result) results.extend(self.postprocess_func(out, image_file_list))
batch_data.clear() batch_data.clear()
image_file_list.clear() image_file_list.clear()
return results
def export(self): def export(self):
assert self.mode == "export" assert self.mode == "export"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册