Skip to content

GitLab

Commit eb96c623 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

set benchmark in predict_system

parent 9b60942b
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
tools/infer/predict_system.py
View file @ eb96c623
...@@ -173,7 +173,9 @@ def main(args): ...@@ -173,7 +173,9 @@ def main(args):
logger.info("The predict total time is {}".format(time.time() - _st)) logger.info("The predict total time is {}".format(time.time() - _st))
logger.info("\nThe predict total time is {}".format(total_time)) 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__": if __name__ == "__main__":
args = utility.parse_args() args = utility.parse_args()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment