diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index eae0e27cd284ccce9f41f0c20b05dee09f46fc84..6e7f69025a924ed727905c7efa4867262da665b7 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()