__init__.py 602 B

12345678910111213141516171819202122232425262728293031323334
  1. """
  2. Vendor 包 - 第三方依赖
  3. 包含从 MinerU 移植的核心模块
  4. """
  5. # 设备检测工具
  6. from .device_utils import get_device, get_device_name
  7. # OCR 工具函数
  8. from .ocr_utils import (
  9. check_img,
  10. preprocess_image,
  11. sorted_boxes,
  12. merge_det_boxes,
  13. update_det_boxes,
  14. get_rotate_crop_image
  15. )
  16. __all__ = [
  17. # 设备工具
  18. 'get_device',
  19. 'get_device_name',
  20. # OCR 工具
  21. 'check_img',
  22. 'preprocess_image',
  23. 'sorted_boxes',
  24. 'merge_det_boxes',
  25. 'update_det_boxes',
  26. 'get_rotate_crop_image',
  27. # OCR 引擎
  28. 'PytorchPaddleOCR',
  29. ]