Sfoglia il codice sorgente

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

zhch158_admin 1 settimana fa
parent
commit
1f38e81a65
1 ha cambiato i file con 3 aggiunte e 1 eliminazioni
  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'