from watermark_verify import verify_tool

if __name__ == '__main__':
    # model_filename = "/mnt/d/WorkSpace/PyCharmGitWorkspace/model_watermark_verify/tests/blackbox_models/vgg16/vgg16.onnx"
    # model_filename = "/mnt/d/WorkSpace/PyCharmGitWorkspace/model_watermark_verify/tests/blackbox_models/alexnet/alexnet.onnx"
    # model_filename = "/mnt/d/WorkSpace/PyCharmGitWorkspace/model_watermark_verify/tests/blackbox_models/resnet/resnet101.onnx"
    # model_filename = "/mnt/d/WorkSpace/PyCharmGitWorkspace/model_watermark_verify/tests/blackbox_models/googlenet/googlenet.onnx"
    model_filename = "/mnt/d/WorkSpace/PyCharmGitWorkspace/model_watermark_verify/tests/blackbox_models/googlenet/googlenet_pruned.onnx"
    verify_result = verify_tool.label_verification(model_filename)
    print(f"verify_result: {verify_result}")