ソースを参照

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

Suven 11 ヶ月 前
コミット
be01039495
1 ファイル変更2 行追加0 行削除
  1. 2 0
      magic_pdf/model/batch_analyze.py

+ 2 - 0
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}")