Method load_ocr_model

ddddocr/models/model_loader.py:119–150

加载OCR模型 Args: old: 是否使用旧版模型 beta: 是否使用beta版模型 import_onnx_path: 自定义模型路径 Returns: ONNX推理会话对象 Raises: ModelLoadError: 当模型加载失败时

(self, old: bool = False, beta: bool = False, 
                      import_onnx_path: str = "")

            return {'error': str(e)}
    
    def load_ocr_model(self, old: bool = False, beta: bool = False, 
                      import_onnx_path: str = "") -> onnxruntime.InferenceSession:
        """
        加载OCR模型
        
        Args:
            old: 是否使用旧版模型
            beta: 是否使用beta版模型
            import_onnx_path: 自定义模型路径
            
        Returns:
            ONNX推理会话对象
            
        Raises:
            ModelLoadError: 当模型加载失败时
        """
        try:
            if import_onnx_path:
                model_path = import_onnx_path
            else:
                base_dir = os.path.dirname(os.path.dirname(__file__))
                if old:
                    model_path = os.path.join(base_dir, 'common_old.onnx')
                elif beta:
                    model_path = os.path.join(base_dir, 'common.onnx')
                else:
                    model_path = os.path.join(base_dir, 'common_old.onnx')
            
            return self.load_model(model_path)
            
        except Exception as e:
            raise ModelLoadError(f"OCR模型加载失败: {str(e)}") from e
    
    def load_detection_model(self) -> onnxruntime.InferenceSession:
        """