From be0103949565bca102868700c19f0d59d5faa5e9 Mon Sep 17 00:00:00 2001 From: Suven Date: Fri, 13 Dec 2024 08:37:11 +0800 Subject: [PATCH] feat: add logging for detection time in BatchAnalyze when OCR is not applied --- magic_pdf/model/batch_analyze.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/magic_pdf/model/batch_analyze.py b/magic_pdf/model/batch_analyze.py index 51adae2b..8795a382 100644 --- a/magic_pdf/model/batch_analyze.py +++ b/magic_pdf/model/batch_analyze.py @@ -191,6 +191,8 @@ class BatchAnalyze: if self.model.apply_ocr: logger.info(f"ocr time: {round(ocr_time, 2)}, image num: {ocr_count}") + else: + logger.info(f"det time: {round(ocr_time, 2)}, image num: {ocr_count}") if self.model.apply_table: logger.info(f"table time: {round(table_time, 2)}, image num: {table_count}")