set benchmark in predict_system

2026-09-24 23:33:08 +08:00 · 2021-08-27 06:46:40 +00:00
parent 9b60942b0c
commit eb96c623b8
1 changed files with 3 additions and 1 deletions
@@ -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()