From 7c7aff46f437aab41c396269133ad8054edca40c Mon Sep 17 00:00:00 2001 From: ngreenwald Date: Thu, 21 May 2020 19:05:42 -0700 Subject: [PATCH] fix indexing error which prevented loading of 2D data --- caliban_toolbox/pre_annotation/data_loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/caliban_toolbox/pre_annotation/data_loader.py b/caliban_toolbox/pre_annotation/data_loader.py index 00409d9..b887dcf 100644 --- 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