Переглянути джерело

修改AlexNet黑盒水印测试代码

liyan 11 місяців тому
батько
коміт
8a361c75fe
1 змінених файлів з 1 додано та 1 видалено
  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}")