michael/modrec-workflow
1
0
Fork 0
forked from qoherent/modrec-workflow
Code Pull Requests Workflows Activity

fixed ort

Browse Source
This commit is contained in:
Liyu Xiao 2025-06-18 14:27:08 -04:00
parent 9f4b71437e
commit 3437512d7c
1 changed files with 5 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
scripts/ort/convert_to_ort.py
View File

@ -4,10 +4,8 @@ from helpers.app_settings import get_app_settings
settings = get_app_settings() settings = get_app_settings()
input_path = "onnx_files/inference_recognition_model.onnx" input_path = "onnx_files/inference_recognition_model.onnx"
optimization_style = settings.app.optimization_style optimization_style = settings.app.optimization_style
target_platform = settings.app.optimization_style target_platform = settings.app.target_platform
# Build the command # Build the command
command = [ command = [
@ -15,13 +13,10 @@ command = [
"-m", "-m",
"onnxruntime.tools.convert_onnx_models_to_ort", "onnxruntime.tools.convert_onnx_models_to_ort",
input_path, input_path,
"--output_dir", "--output_dir", "ort_files",
"ort_files", "--optimization_style", optimization_style,
"--optimization_style", "--target_platform", target_platform,
optimization_style,
"--target_platform",
target_platform,
] ]
# Run it # Run the command
subprocess.run(command, check=True) subprocess.run(command, check=True)