From 9f621279b8a66672e22e0c5245a4708a3bdc5e3d Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Mon, 17 Apr 2023 12:07:53 +0000
Subject: [PATCH] fix infer output

---
 ppcls/engine/engine.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py
index 40496ab0..378d241b 100755
--- a/ppcls/engine/engine.py
+++ b/ppcls/engine/engine.py
@@ -522,7 +522,9 @@ class Engine(object):
                 if isinstance(out, dict) and "output" in out:
                     out = out["output"]
 
-                results.extend(self.postprocess_func(out, image_file_list))
+                result = self.postprocess_func(out, image_file_list)
+                logger.info(result)
+                results.extend(result)
                 batch_data.clear()
                 image_file_list.clear()
         return results
-- 
GitLab