fix indexing error which prevented loading of 2D data

vanvalenlab · May 22, 2020 · 7c7aff4 · 7c7aff4
1 parent a012476
commit 7c7aff4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/caliban_toolbox/pre_annotation/data_loader.py b/caliban_toolbox/pre_annotation/data_loader.py
@@ -428,7 +428,7 @@ def load_imagedata(self):
                     # Read in the image
                     img_set = tiff.imread(path)
                     raw_images.append(img_set)
-                    if img_set.shape[0] > max_frames:
+                    if img_set.shape[0] > max_frames and len(img_set.shape) > 2:
                         max_frames = img_set.shape[0]
 
         # TODO: the following wont be neccesary when num_frames exist