Class ModelWrapper

modules/base.py:184–216

模型封装基类 为了让"调用模型时应该就是一句话"，我们将模型封装成简单的接口。 使用示例: model = SAM3Model() result = model.predict(image, prompts) # 一行调用

class ModelWrapper:
    """
    模型封装基类
    
    为了让"调用模型时应该就是一句话"，我们将模型封装成简单的接口。
    
    使用示例:
        model = SAM3Model()
        result = model.predict(image, prompts)  # 一行调用
    """
    
    def __init__(self):
        self._model = None
        self._is_loaded = False
    
    @abstractmethod
    def load(self):
        """加载模型"""
        pass
    
    @abstractmethod
    def predict(self, *args, **kwargs):
        """模型推理"""
        pass
    
    def unload(self):
        """卸载模型"""
        self._model = None
        self._is_loaded = False
    
    @property
    def is_loaded(self) -> bool:
        return self._is_loaded