fix: ZCYX doesn't throw error but corrected to CZYX

kreshuklab · Dec 19, 2024 · cfb9c14 · cfb9c14
1 parent a01e65c
commit cfb9c14
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/plantseg/core/image.py b/plantseg/core/image.py
@@ -417,7 +417,10 @@ def _check_shape(self, data: np.ndarray, properties: ImageProperties) -> tuple[n
                 return data[:, 0], properties
 
         elif self.image_layout == ImageLayout.ZCYX:
-            raise ValueError(f"Image layout {self.image_layout} not supported, should have been converted to CZYX")
+            logger.warning("Image layout is ZCYX but should have been converted to CZYX. PlantSeg is doing this now.")
+            properties.image_layout = ImageLayout.CZYX
+            data = np.moveaxis(data, 0, 1)
+            return self._check_shape(data, properties)
 
         return data, properties