ocr_platform/ocr_tools/pytorch_models/__init__.py

"""
PyTorch 模型工具包

提供模型转换（Paddle → ONNX/PyTorch）和 PyTorch 推理功能：
- OCR 识别（检测 + 识别）
- 布局检测
- 文档方向分类
- 文档预处理 Pipeline
"""

# 添加 ocr_platform 根目录到 Python 路径（用于导入 ocr_utils）
import sys
from pathlib import Path

ocr_platform_root = Path(__file__).parents[2]  # pytorch_models -> ocr_tools -> ocr_platform -> repository.git
if str(ocr_platform_root) not in sys.path:
    sys.path.insert(0, str(ocr_platform_root))

# 导入主要类和函数
from .pytorch_paddle import PytorchPaddleOCR
from .layout_detect_onnx import LayoutDetectorONNX
from .orientation_classifier_v2 import OrientationClassifierV2, OrientationResult
from .doc_preprocessor_v2 import DocPreprocessorV2
from .doc_preprocess_result import DocPreprocessResult
from .paddle_to_pytorch_universal import UniversalPaddleToPyTorchConverter, PaddleModelFormat
from .unified_model_loader import UnifiedModelLoader

__all__ = [
    'PytorchPaddleOCR',
    'LayoutDetectorONNX',
    'OrientationClassifierV2',
    'OrientationResult',
    'DocPreprocessorV2',
    'DocPreprocessResult',
    'UniversalPaddleToPyTorchConverter',
    'PaddleModelFormat',
    'UnifiedModelLoader',
]

__version__ = "1.0.0"