diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py index ac7534dd506381a37fe03d6dcfad6607b9f81f06..12d183674d16684d528573d709fc5fc5730ada41 100755 --- a/ppcls/engine/engine.py +++ b/ppcls/engine/engine.py @@ -520,10 +520,11 @@ class Engine(object): out = out["logits"] if isinstance(out, dict) and "output" in out: 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() image_file_list.clear() + return results def export(self): assert self.mode == "export"