introduce new feature: GenRefLabels 3

tum-ei-eda · May 27, 2024 · 7b435d9 · 7b435d9
1 parent 86d7aa9
commit 7b435d9
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/mlonmcu/models/model.py b/mlonmcu/models/model.py
@@ -163,6 +163,7 @@ class Model(Workload):
         "support_path": None,
         "inputs_path": None,
         "outputs_path": None,
+        "output_labels_path": None,
     }
 
     def __init__(self, name, paths, config=None, alt=None, formats=ModelFormats.TFLITE):
@@ -248,6 +249,16 @@ def outputs_path(self):
                 value = Path(value)
         return value
 
+    @property
+    def output_labels_path(self):
+        # TODO: fall back to metadata
+        value = self.config["output_labels_path"]
+        if value is not None:
+            if not isinstance(value, Path):
+                assert isinstance(value, str)
+                value = Path(value)
+        return value
+
     @property
     def skip_check(self):
         if len(self.formats) == 0: