set benchmark in predict_system

This commit is contained in:
LDOUBLEV 2021-08-27 06:46:40 +00:00
parent 9b60942b0c
commit eb96c623b8
1 changed files with 3 additions and 1 deletions

View File

@ -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()