Skip to content

GitLab

Commit be010394 authored by Suven's avatar Suven
Browse files

feat: add logging for detection time in BatchAnalyze when OCR is not applied

parent 49bfdf07
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
magic_pdf/model/batch_analyze.py
View file @ be010394
...@@ -191,6 +191,8 @@ class BatchAnalyze: ...@@ -191,6 +191,8 @@ class BatchAnalyze:
if self.model.apply_ocr: if self.model.apply_ocr:
logger.info(f"ocr time: {round(ocr_time, 2)}, image num: {ocr_count}") 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: if self.model.apply_table:
logger.info(f"table time: {round(table_time, 2)}, image num: {table_count}") logger.info(f"table time: {round(table_time, 2)}, image num: {table_count}")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment