From 3437512d7cad13feee82a3949d210d95ef66e4c2 Mon Sep 17 00:00:00 2001 From: Liyu Xiao Date: Wed, 18 Jun 2025 14:27:08 -0400 Subject: [PATCH] fixed ort --- scripts/ort/convert_to_ort.py | 15 +++++---------- 1 file changed, 5 insertions(+), 10 deletions(-) diff --git a/scripts/ort/convert_to_ort.py b/scripts/ort/convert_to_ort.py index 8d44f5a..4eaa9d8 100644 --- a/scripts/ort/convert_to_ort.py +++ b/scripts/ort/convert_to_ort.py @@ -4,10 +4,8 @@ from helpers.app_settings import get_app_settings settings = get_app_settings() input_path = "onnx_files/inference_recognition_model.onnx" - - optimization_style = settings.app.optimization_style -target_platform = settings.app.optimization_style +target_platform = settings.app.target_platform # Build the command command = [ @@ -15,13 +13,10 @@ command = [ "-m", "onnxruntime.tools.convert_onnx_models_to_ort", input_path, - "--output_dir", - "ort_files", - "--optimization_style", - optimization_style, - "--target_platform", - target_platform, + "--output_dir", "ort_files", + "--optimization_style", optimization_style, + "--target_platform", target_platform, ] -# Run it +# Run the command subprocess.run(command, check=True)