瀏覽代碼

feat(model): add logging for batch image processing

- Add logger info for each batch processed
- Include batch number and page count in log message
myhloli 6 月之前
父節點
當前提交
b29b73af21
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      magic_pdf/model/doc_analyze_by_custom_model.py

+ 4 - 1
magic_pdf/model/doc_analyze_by_custom_model.py

@@ -156,7 +156,10 @@ def doc_analyze(
         batch_images = [images_with_extra_info]
         batch_images = [images_with_extra_info]
 
 
     results = []
     results = []
-    for batch_image in batch_images:
+    processed_images_count = 0
+    for index, batch_image in enumerate(batch_images):
+        processed_images_count += len(batch_image)
+        logger.info(f'Batch {index + 1}/{len(batch_images)}: {processed_images_count} pages/{len(images_with_extra_info)} pages')
         result = may_batch_image_analyze(batch_image, ocr, show_log,layout_model, formula_enable, table_enable)
         result = may_batch_image_analyze(batch_image, ocr, show_log,layout_model, formula_enable, table_enable)
         results.extend(result)
         results.extend(result)