Only determine segments for single-channel images

DIAGNijmegen · Jul 17, 2023 · 10712d1 · 10712d1
1 parent 0b25efb
commit 10712d1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/panimg/models.py b/panimg/models.py
@@ -264,7 +264,12 @@ def add_value_range_meta_data(cls, image: Image):  # noqa: B902, N805
 
     @property
     def segments(self) -> Optional[FrozenSet[int]]:
-        if self.image.GetPixelIDValue() not in MASK_TYPE_PIXEL_IDS:
+        if (
+            self.image.GetNumberOfComponentsPerPixel() != 1
+            or self.image.GetPixelIDValue() not in MASK_TYPE_PIXEL_IDS
+        ):
+            # Only single channel, 8 bit images should be checked.
+            # Everything else is not a segmentation
             return None
 
         im_arr = GetArrayViewFromImage(self.image)