fix save_log_path as null

7ad66311 · LDOUBLEV · d93a445d · 7ad66311 · 7ad66311
显示空白变更内容
内联并排

Showing with 4 addition and 2 deletion

tools/infer/predict_det.py tools/infer/predict_det.py +2 -1

tools/infer/predict_rec.py tools/infer/predict_rec.py +2 -1

未找到文件。
--- 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")
--- 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")