imx-machine-learning: check vela return code
Return an error in case any of the models could not be converted. https://onedigi.atlassian.net/browse/DEL-9424 Signed-off-by: Isaac Hermida <isaac.hermida@digi.com>
This commit is contained in:
parent
25f4fa7641
commit
8ad093c37b
|
|
@ -0,0 +1,26 @@
|
||||||
|
From: Isaac Hermida <isaac.hermida@digi.com>
|
||||||
|
Date: Tue, 17 Dec 2024 18:34:48 +0100
|
||||||
|
Subject: [PATCH] check vela return code
|
||||||
|
|
||||||
|
Signed-off-by: Isaac Hermida <isaac.hermida@digi.com>
|
||||||
|
---
|
||||||
|
download_models.py | 6 +++++-
|
||||||
|
1 file changed, 5 insertions(+), 1 deletion(-)
|
||||||
|
|
||||||
|
diff --git a/download_models.py b/download_models.py
|
||||||
|
index 462f7b755eea..fc5ab234ae47 100644
|
||||||
|
--- a/download_models.py
|
||||||
|
+++ b/download_models.py
|
||||||
|
@@ -103,7 +103,11 @@ def convert_model(model_dir, vela_dir):
|
||||||
|
if name.endswith(".tflite"):
|
||||||
|
print('Converting', name)
|
||||||
|
model = os.path.join(model_dir, name)
|
||||||
|
- os.system('vela ' + model + " --output-dir " + vela_dir)
|
||||||
|
+ ret = os.system('vela ' + model + " --output-dir " + vela_dir)
|
||||||
|
+ if ret != 0:
|
||||||
|
+ print(f"Error: Failed to convert model {name} with vela")
|
||||||
|
+ import sys
|
||||||
|
+ sys.exit(1)
|
||||||
|
|
||||||
|
model_dir = 'models'
|
||||||
|
vela_dir = 'vela_models'
|
||||||
|
|
@ -11,6 +11,7 @@ VELA_MODELS_DIR = "vela_models"
|
||||||
SRC_URI += " \
|
SRC_URI += " \
|
||||||
file://0001-Customize-EiQ-demos.patch \
|
file://0001-Customize-EiQ-demos.patch \
|
||||||
file://0002-improvements-capture-x-windows-and-increase-resoluti.patch \
|
file://0002-improvements-capture-x-windows-and-increase-resoluti.patch \
|
||||||
|
file://0003-check-vela-return-code.patch \
|
||||||
file://scripts/launch_eiq_demo.sh \
|
file://scripts/launch_eiq_demo.sh \
|
||||||
file://service/eiqdemo.service \
|
file://service/eiqdemo.service \
|
||||||
"
|
"
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue