PaddleX/zhch/utils/__init__.py

"""
工具模块包

包含CUDA环境检测、文件处理等实用工具函数
"""

from .check_cuda_env import (
    check_nvidia_environment,
)

from .cuda_utils import (
    monitor_gpu_memory,
    detect_available_gpus
)

from .file_utils import (
    get_image_files_from_dir,
    get_image_files_from_list,
    get_image_files_from_csv,
    split_files,
    create_temp_file_list,
    collect_pid_files,
)

from .doc_utils import (
    load_images_from_pdf,
    fitz_doc_to_image,
)

from .normalize_financial_numbers import (
    normalize_financial_numbers,
    normalize_markdown_table,
    normalize_json_table,
)

__all__ = [
    # CUDA环境检测
    'check_nvidia_environment',
    # CUDA工具
    'monitor_gpu_memory',
    'detect_available_gpus',
    # 文件工具
    'get_image_files_from_dir',
    'get_image_files_from_list',
    'get_image_files_from_csv',
    'split_files',
    'create_temp_file_list',
    'collect_pid_files',
    # 金融数字标准化
    'normalize_financial_numbers',
    'normalize_markdown_table',
    'normalize_json_table',
]

__version__ = "1.0.0"
__author__ = "zhch158"
__description__ = "PaddleX工具模块包，提供CUDA环境检测和文件处理功能"