Method load_detection_model

ddddocr/models/model_loader.py:152–168

加载目标检测模型 Returns: ONNX推理会话对象 Raises: ModelLoadError: 当模型加载失败时

(self)

            raise ModelLoadError(f"OCR模型加载失败: {str(e)}") from e
    
    def load_detection_model(self) -> onnxruntime.InferenceSession:
        """
        加载目标检测模型
        
        Returns:
            ONNX推理会话对象
            
        Raises:
            ModelLoadError: 当模型加载失败时
        """
        try:
            base_dir = os.path.dirname(os.path.dirname(__file__))
            model_path = os.path.join(base_dir, 'common_det.onnx')
            return self.load_model(model_path)
            
        except Exception as e:
            raise ModelLoadError(f"检测模型加载失败: {str(e)}") from e
    
    def load_custom_model(self, model_path: str, charset_path: str) -> tuple:
        """