loading multipage tifs in a project

kushalkolar · Mar 25, 2021 · a5545f5 · a5545f5
1 parent fdba1df
commit a5545f5
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/mesmerize/viewer/core/viewer_work_environment.py b/mesmerize/viewer/core/viewer_work_environment.py
@@ -160,7 +160,13 @@ def from_pickle(cls, pickle_file_path: str, tiff_path: str = None):
         if tiff_path is not None:
             # seq = tifffile.imread(tiff_path)
             tif = tifffile.TiffFile(tiff_path, is_nih=True)
-            seq = tif.asarray(maxworkers=int(get_sys_config()['_MESMERIZE_N_THREADS']))
+            if len(tif.series) > 1:
+                seq = tif.asarray(
+                    key=range(0, len(tif.series)),
+                    maxworkers=int(get_sys_config()['_MESMERIZE_N_THREADS'])
+                )
+            else:
+                seq = tif.asarray(maxworkers=int(get_sys_config()['_MESMERIZE_N_THREADS']))
             if seq.ndim == 4:
                 # tzxy to xytz
                 seq = np.moveaxis(seq, (0, 1, 2, 3), (2, 3, 0, 1))