mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

Fixed the incorrect infer outputs

Browse Source
This commit is contained in:
parap1uie-s 2023-03-17 17:46:11 +08:00 committed by Walter
parent 57504f2648
commit 6e6586f59b
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ppcls/engine/engine.py
View File

@ -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"