from watermark_verify.process import classification_pytorch_blackbox_process


def label_verification(model_filename: str) -> bool:
    """
    模型标签提取验证
    :param model_filename: 模型权重文件，onnx格式
    :return: 模型标签验证结果
    """
    # 初始化模型水印检测器
    model_detector = classification_pytorch_blackbox_process.ModelWatermarkProcessor(model_filename)
    result = model_detector.process()  # 获取模型水印检测结果
    return result