浏览代码

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'