Merge pull request #514 from openvinotoolkit/fix-instance-segmentation-label-id-parsing

Fix empty label assignment for instance segmentation models

Author: ljcornel

geti_sdk/data_models/containers/label_list.py

@@ -124,5 +124,11 @@ def sort_by_ids(self, label_ids: List[str]):
         """
         new_data: List[Label] = []
         for label_id in label_ids:
+            if label_id is None:
+                # Certain models have the label id as 'None' to signal an empty label
+                if self.get_empty_label() is not None:
+                    new_data.append(self.get_empty_label())
+                else:
+                    continue
             new_data.append(self.get_by_id(label_id))
         self.data = new_data

geti_sdk/deployment/deployed_model.py

@@ -764,6 +764,9 @@ def get_model_config(self) -> Dict[str, Any]:
                 value = value.split(" ")
                 value_list = []
                 for item in value:
+                    if isinstance(item, str) and item.lower() == "none":
+                        value_list.append(None)
+                        continue
                     try:
                         value_list.append(float(item))
                     except ValueError: