Method load_custom_model

ddddocr/models/model_loader.py:170–204

加载自定义模型和字符集 Args: model_path: 模型文件路径 charset_path: 字符集文件路径 Returns: (推理会话, 字符集信息) Raises: ModelLoadError: 当加载失败时

(self, model_path: str, charset_path: str)

            raise ModelLoadError(f"检测模型加载失败: {str(e)}") from e
    
    def load_custom_model(self, model_path: str, charset_path: str) -> tuple:
        """
        加载自定义模型和字符集
        
        Args:
            model_path: 模型文件路径
            charset_path: 字符集文件路径
            
        Returns:
            (推理会话, 字符集信息)
            
        Raises:
            ModelLoadError: 当加载失败时
        """
        try:
            # 加载模型
            session = self.load_model(model_path)
            
            # 加载字符集信息
            if not os.path.exists(charset_path):
                raise ModelLoadError(f"字符集文件不存在: {charset_path}")
            
            with open(charset_path, 'r', encoding="utf-8") as f:
                charset_info = json.loads(f.read())
            
            # 验证字符集信息格式
            required_keys = ['charset', 'word', 'image', 'channel']
            for key in required_keys:
                if key not in charset_info:
                    raise ModelLoadError(f"字符集文件缺少必需字段: {key}")
            
            return session, charset_info
            
        except Exception as e:
            raise ModelLoadError(f"自定义模型加载失败: {str(e)}") from e
    
    def validate_model_compatibility(self, session: onnxruntime.InferenceSession, 
                                   expected_input_shape: Optional[List[int]] = None) -> bool: