From 6e6586f59bb22bd033725fbb589f957c75654d73 Mon Sep 17 00:00:00 2001
From: parap1uie-s <parap1uie-s@users.noreply.github.com>
Date: Fri, 17 Mar 2023 17:46:11 +0800
Subject: [PATCH] Fixed the incorrect infer outputs

---
 ppcls/engine/engine.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py
index ac7534dd..12d18367 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"
-- 
GitLab