forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix save_log_path as null

This commit is contained in:
LDOUBLEV 2021-06-09 18:20:36 +08:00
parent d93a445d4c
commit 7ad663119a
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/infer/predict_det.py
View File

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

3
tools/infer/predict_rec.py
View File

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