fix(io/heyex/xml_export): initalize empty LayerAnnotation if no annot…

…ation is provided
MedVisBonn · Aug 16, 2021 · 6626467 · 6626467
1 parent 54e0a6c
commit 6626467
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/eyepy/io/heyex/xml_export.py b/eyepy/io/heyex/xml_export.py
@@ -152,8 +152,11 @@ def layers_dict(bscan_obj):
                         float(x) for x in segline.find("./Array").text.split()
                     ]
                 return LayerAnnotation(data, max_height=bscan_obj.oct_obj.SizeZ)
+            else:
+                warnings.warn(f"{bscan_obj} contains no segmentation", UserWarning)
+                data = np.zeros((max(config.SEG_MAPPING.values()) + 1, self.oct_obj.SizeX))
+                return LayerAnnotation(data, max_height=bscan_obj.oct_obj.SizeZ)
 
-            warnings.warn(f"{bscan_obj} contains no segmentation", UserWarning)
 
         return {
             "layers": layers_dict,