Method __init__

tools/infer_rec.py:195–265

Args: config (dict, optional): 配置信息。默认为None。 mode (str, optional): 模式，'server' 或 'mobile'。默认为'mobile'。 backend (str): 'torch' 或 'onnx' onnx_model_path (str): ONNX模型路径（仅当backend='onnx'时需要） use_gpu (str, optional): GPU使用策略，可选值为'auto'

(self,
                 config=None,
                 mode='mobile',
                 backend='torch',
                 onnx_model_path=None,
                 use_gpu='auto',
                 numId=0)

class OpenRecognizer:

    def __init__(self,
                 config=None,
                 mode='mobile',
                 backend='torch',
                 onnx_model_path=None,
                 use_gpu='auto',
                 numId=0):
        """
        Args:
            config (dict, optional): 配置信息。默认为None。
            mode (str, optional): 模式，'server' 或 'mobile'。默认为'mobile'。
            backend (str): 'torch' 或 'onnx'
            onnx_model_path (str): ONNX模型路径（仅当backend='onnx'时需要）
            use_gpu (str, optional): GPU使用策略，可选值为'auto'/'true'/'false'。默认为'auto'。
            numId (int, optional): 设备编号。默认为0。
        """

        # Auto-switch backend for server mode
        if mode == 'server' and backend != 'torch':
            logger.warning(
                f"Server mode only supports 'torch' backend, got '{backend}'. "
                f"Automatically switching to 'torch' backend. "
                f"Please make sure 'torch' and 'torchvision' are installed: "
                f"pip install torch torchvision")
            backend = 'torch'

        if config is None:
            config_file = DEFAULT_CFG_PATH_REC_SERVER if mode == 'server' else DEFAULT_CFG_PATH_REC
            config = Config(config_file).cfg

        # Parse use_gpu parameter
        if use_gpu == 'auto':
            try:
                import torch
                device = 'gpu' if torch.cuda.is_available() else 'cpu'
            except:
                device = 'cpu'
        elif use_gpu == 'true':
            device = 'gpu'
        elif use_gpu == 'false':
            device = 'cpu'
        else:
            raise ValueError(f"use_gpu must be 'auto', 'true', or 'false', got '{use_gpu}'")

        config['Global']['device'] = device

        self.cfg = config
        # 公共初始化
        self._init_common()
        backend = backend if config['Global'].get(
            'backend', None) is None else config['Global']['backend']
        self.backend = backend
        if backend == 'torch':
            import torch
            self.torch = torch
            self._init_torch_model(numId)
        elif backend == 'onnx':
            from tools.infer.onnx_engine import ONNXEngine