Jelajahi Sumber

Merge pull request #2410 from myhloli/dev

feat(model): add logging for batch image processing
Xiaomeng Zhao 6 bulan lalu
induk
melakukan
f0b66d3aab
1 mengubah file dengan 4 tambahan dan 1 penghapusan
  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]
 
     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)
         results.extend(result)