From 7ad663119aa35e9fb28ebab505c19dccb057e8cc Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Wed, 9 Jun 2021 18:20:36 +0800 Subject: [PATCH] fix save_log_path as null --- tools/infer/predict_det.py | 3 ++- tools/infer/predict_rec.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index baa89be1..2463fdf3 100755 --- a/tools/infer/predict_det.py +++ b/tools/infer/predict_det.py @@ -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") diff --git a/tools/infer/predict_rec.py b/tools/infer/predict_rec.py index 2eeb39b2..c0d6ef68 100755 --- a/tools/infer/predict_rec.py +++ b/tools/infer/predict_rec.py @@ -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") -- GitLab