|
|
@@ -4,6 +4,6 @@
|
|
|
from watermark_verify import verify_tool
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
- model_filename = "/mnt/e/DevTools/workspace/model_watermark_detect/tests/models/blackbox/alexnet/alexnet.onnx"
|
|
|
+ model_filename = "/home/ai/checkpoints/pytorch-alexnet-blackbox/om/model_99.om"
|
|
|
verify_result = verify_tool.label_verification(model_filename)
|
|
|
print(f"verify_result: {verify_result}")
|
