fix infer output

9f621279 · gaotingquan · Tingquan Gao · 73f4d8e4 · 9f621279
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

ppcls/engine/engine.py ppcls/engine/engine.py +3 -1

未找到文件。
--- 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