model_watermark_verify/tests/verify_tool_test.py

from watermark_verify import verify_tool

if __name__ == '__main__':
    # model_filename = "yolox_s.onnx"
    # verify_result = verify_tool.label_verification(model_filename)
    # print(f"verify_result: {verify_result}")
    # test ssd model
    model_filename = "faster-rcnn.onnx"
    verify_result = verify_tool.label_verification(model_filename)
    print(f"verify_result: {verify_result}")