Prechádzať zdrojové kódy

feat: 新增PaddleVL识别器支持,更新获取逻辑以适配不同模块

zhch158_admin 2 týždňov pred
rodič
commit
b0dea4a516

+ 5 - 2
zhch/universal_doc_parser/models/adapters/__init__.py

@@ -12,6 +12,7 @@ from .base import (
 )
 
 from .paddle_layout_detector import PaddleLayoutDetector
+from .paddle_vl_adapter import PaddleVLRecognizer
 
 # 可选导入 MinerU 适配器
 try:
@@ -35,6 +36,7 @@ __all__ = [
     
     # PaddleX 适配器
     'PaddleLayoutDetector',
+    'PaddleVLRecognizer',
 ]
 
 # 如果 MinerU 可用,添加到导出列表
@@ -85,14 +87,15 @@ def get_vl_recognizer(config: dict):
     """根据配置获取VL识别器"""
     module = config.get('module', 'mineru')
     
-    if module == 'mineru':
+    if module == 'paddle':
+        return PaddleVLRecognizer(config)
+    elif module == 'mineru':
         if not MINERU_AVAILABLE:
             raise ImportError("MinerU adapter not available")
         return MinerUVLRecognizer(config)
     else:
         raise ValueError(f"Unknown VL recognizer module: {module}")
 
-
 def get_ocr_recognizer(config: dict):
     """根据配置获取OCR识别器"""
     module = config.get('module', 'mineru')