|
|
@@ -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')
|