PaddleX/zhch/unified_pytorch_models/vendor/__init__.py

"""
Vendor 包 - 第三方依赖
包含从 MinerU 移植的核心模块
"""

# 设备检测工具
from .device_utils import get_device, get_device_name

# OCR 工具函数
from .ocr_utils import (
    check_img,
    preprocess_image,
    sorted_boxes,
    merge_det_boxes,
    update_det_boxes,
    get_rotate_crop_image
)

# PytorchPaddleOCR 主类
from .pytorch_paddle import PytorchPaddleOCR

__all__ = [
    # 设备工具
    'get_device',
    'get_device_name',
    
    # OCR 工具
    'check_img',
    'preprocess_image',
    'sorted_boxes',
    'merge_det_boxes',
    'update_det_boxes',
    'get_rotate_crop_image',
    
    # OCR 引擎
    'PytorchPaddleOCR',
]