model_watermark_verify/tests/yolox_inference_test.py

from watermark_verify.inference import yolox


if __name__ == '__main__':

    test_img = "000000000030.jpg"
    model_file = "yolox_s.onnx"
    output_dir = "./output"
    watermark_txt = "./trigger/qrcode_positions.txt"
    input_shape = (640, 640)
    detect_result = yolox.predict_and_detect(test_img, model_file, watermark_txt, input_shape)
    print(f"detect_result={detect_result}")