Quellcode durchsuchen

fix: enhance table prediction logic by incorporating table classification score and refining model selection criteria

myhloli vor 3 Monaten
Ursprung
Commit
706eadbf5d
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. 1 1
      mineru/backend/pipeline/batch_analyze.py

+ 1 - 1
mineru/backend/pipeline/batch_analyze.py

@@ -10,7 +10,7 @@ from ...utils.config_reader import get_formula_enable, get_table_enable
 from ...utils.model_utils import crop_img, get_res_list_from_layout_res
 from ...utils.ocr_utils import get_adjusted_mfdetrec_res, get_ocr_result_list, OcrConfidence
 
-YOLO_LAYOUT_BASE_BATCH_SIZE = 8
+YOLO_LAYOUT_BASE_BATCH_SIZE = 1
 MFD_BASE_BATCH_SIZE = 1
 MFR_BASE_BATCH_SIZE = 16
 OCR_DET_BASE_BATCH_SIZE = 16