From eb96c623b8d2a6f3b6e973c76f5b0349546f36bc Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Fri, 27 Aug 2021 06:46:40 +0000
Subject: [PATCH] set benchmark in predict_system

---
 tools/infer/predict_system.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py
index eae0e27c..6e7f6902 100755
--- a/tools/infer/predict_system.py
+++ b/tools/infer/predict_system.py
@@ -173,7 +173,9 @@ def main(args):
 
     logger.info("The predict total time is {}".format(time.time() - _st))
     logger.info("\nThe predict total time is {}".format(total_time))
-
+    if args.benchmark:
+        text_sys.text_detector.autolog.report()
+        text_sys.text_recognizer.autolog.report()
 
 if __name__ == "__main__":
     args = utility.parse_args()
-- 
GitLab