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