Преглед на файлове

fix: 修正 DotsOCR 的合并脚本映射,确保正确使用合并脚本

zhch158_admin преди 1 седмица
родител
ревизия
1f38e81a65
променени са 1 файла, в които са добавени 3 реда и са изтрити 1 реда
  1. 3 1
      batch_ocr/batch_merge_results.py

+ 3 - 1
batch_ocr/batch_merge_results.py

@@ -40,7 +40,7 @@ class BatchMerger:
     MERGER_SCRIPTS = {
         'paddleocr_vl': 'merge_paddleocr_vl_paddleocr.py',
         'mineru': 'merge_mineru_paddle_ocr.py',
-        'dotsocr': 'merge_mineru_paddle_ocr.py',  # DotsOCR 也用 MinerU 格式
+        'dotsocr': 'merge_dotsocr_paddleocr.py'
     }
     
     def __init__(self, config_file: str, base_dir: str = None):
@@ -405,6 +405,8 @@ class BatchMerger:
             return 'paddleocr-vl'
         elif 'mineru' in script_name:
             return 'mineru'
+        elif 'dotsocr' in script_name:
+            return 'dotsocr'
         else:
             return 'vl'