defect-analysis/defect_analysis/ml/image_models.py

"""图像模型接口占位。

当前项目尚未接入 AOI 图片路径、标注数据和深度学习运行环境，因此这里提供
明确的接口与不可用错误，避免把 PyTorch/TensorFlow 作为硬依赖拖入主应用。
"""


class ImageModelUnavailable(RuntimeError):
    """图像模型后端或模型文件不可用。"""


class ImageModelWrapper:
    """未来 AOI 图像模型的统一包装接口。"""

    def __init__(self, backend=None, model_path=None):
        self.backend = backend
        self.model_path = model_path

    def predict(self, image_paths):
        raise ImageModelUnavailable(
            "图像模型尚未配置。请先接入 AOI 图片路径、标注数据、模型文件和 GPU/推理环境。"
        )