model_watermark_detect/tests/verify_tool_test.py

"""
支持所有待测模型，测试模型水印提取功能，对提供的指定模型文件进行水印检测
"""
import os
import sys
rootpath = str(os.path.abspath(os.path.join(os.path.dirname(__file__), '../')))
sys.path.append(rootpath)
from watermark_verify import verify_tool

if __name__ == '__main__':
    model_filename = "/mnt/e/DevTools/workspace/model_watermark_detect/tests/models/blackbox/yolox/yolox.onnx"
    verify_result = verify_tool.label_verification(model_filename)
    print(f"verify_result: {verify_result}")