提交 7ad66311 编写于 作者: L LDOUBLEV

fix save_log_path as null

上级 d93a445d
......@@ -283,5 +283,6 @@ if __name__ == "__main__":
'total_time_s': det_time_dict['total_time']
}
benchmark_log = benchmark_utils.PaddleInferBenchmark(
text_detector.config, model_info, data_info, perf_info, mems)
text_detector.config, model_info, data_info, perf_info, mems,
args.save_log_path)
benchmark_log("Det")
......@@ -320,7 +320,8 @@ def main(args):
'total_time_s': rec_time_dict['total_time']
}
benchmark_log = benchmark_utils.PaddleInferBenchmark(
text_recognizer.config, model_info, data_info, perf_info, mems)
text_recognizer.config, model_info, data_info, perf_info, mems,
args.save_log_path)
benchmark_log("Rec")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册