Browse Source

feat: 添加 MinerU 和 DotsOCR vLLM 处理器配置,支持多种输入输出参数

zhch158_admin 3 weeks ago
parent
commit
2ca12d8b3e
1 changed files with 28 additions and 0 deletions
  1. 28 0
      zhch/batch_process_pdf.py

+ 28 - 0
zhch/batch_process_pdf.py

@@ -83,6 +83,34 @@ class ConfigManager:
                 ],
                 'output_subdir': 'ppstructurev3_client_results',
                 'description': 'PP-StructureV3 HTTP API 客户端'
+            },
+            'mineru_vllm': {
+                'script': '/Users/zhch158/workspace/repository.git/MinerU/zhch/mineru2_vllm_multthreads.py',
+                'input_arg': '--input_file',
+                'output_arg': '--output_dir',
+                'extra_args': [
+                    '--server_url=http://10.192.72.11:8121',
+                    '--timeout=300',
+                    '--batch_size=1'
+                ],
+                'output_subdir': 'mineru_vllm_results',
+                'description': 'MinerU vLLM 处理器'
+            },
+            'dotsocr_vllm': {
+                'script': '/Users/zhch158/workspace/repository.git/dots.ocr/zhch/dotsocr_vllm_multthreads.py',
+                'input_arg': '--input_file',
+                'output_arg': '--output_dir',
+                'extra_args': [
+                    '--ip=10.192.72.11',
+                    '--port=8101',
+                    '--model_name=DotsOCR',
+                    '--prompt_mode=prompt_layout_all_en',
+                    '--batch_size=1',
+                    '--max_workers=1',
+                    '--dpi=200'
+                ],
+                'output_subdir': 'dotsocr_vllm_results',
+                'description': 'DotsOCR vLLM 处理器 - 支持PDF和图片'
             }
         },
         'global': {