소스 검색

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}")