fixed is_image_rgb function

juglab · Jun 26, 2024 · 8ddd5c1 · 8ddd5c1
1 parent da181ed
commit 8ddd5c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/featureforest/utils/data.py b/src/featureforest/utils/data.py
@@ -247,7 +247,7 @@ def is_image_rgb(image_data: ndarray) -> bool:
     Returns:
         bool: is image RGB(A)?
     """
-    return image_data.shape[-1] >= 3
+    return image_data.shape[-1] in [3, 4]
 
 
 def is_stacked(image_data: ndarray) -> bool: