|
|
@@ -1,4 +1,4 @@
|
|
|
-from watermark_verify.process import classification_pytorch_blackbox_process
|
|
|
+from watermark_verify.process import yolox_pytorch_blackbox_process
|
|
|
|
|
|
|
|
|
def label_verification(model_filename: str) -> bool:
|
|
|
@@ -8,7 +8,7 @@ def label_verification(model_filename: str) -> bool:
|
|
|
:return: 模型标签验证结果
|
|
|
"""
|
|
|
# 初始化模型水印检测器
|
|
|
- model_detector = classification_pytorch_blackbox_process.ModelWatermarkProcessor(model_filename)
|
|
|
+ model_detector = yolox_pytorch_blackbox_process.DetectionProcess(model_filename)
|
|
|
result = model_detector.process() # 获取模型水印检测结果
|
|
|
return result
|
|
|
|
