Selaa lähdekoodia

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

zhch158_admin 1 viikko sitten
vanhempi
commit
1f38e81a65
1 muutettua tiedostoa jossa 3 lisäystä ja 1 poistoa
  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'