model_watermark_detect/tests/verify_tool_test.py

"""
支持所有待测模型，测试模型水印提取功能，对提供的指定模型文件进行水印检测
"""
from watermark_verify import verify_tool

if __name__ == '__main__':
    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}")