Jelajahi Sumber

修改AlexNet黑盒水印测试代码

liyan 11 bulan lalu
induk
melakukan
8a361c75fe
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  1. 1 1
      tests/verify_tool_test.py

+ 1 - 1
tests/verify_tool_test.py

@@ -1,6 +1,6 @@
 from watermark_verify import verify_tool
 
 if __name__ == '__main__':
-    model_filename = "/mnt/d/WorkSpace/PyCharmGitWorkspace/model_watermark_verify/tests/test.onnx"
+    model_filename = "/mnt/d/WorkSpace/PyCharmGitWorkspace/model_watermark_verify/tests/alex.onnx"
     verify_result = verify_tool.label_verification(model_filename)
     print(f"verify_result: {verify_result}")