fixing loading from txt file

hotfix for brainglobe/cellfinder#184
npeschke · Sep 15, 2021 · 79a0107 · 79a0107
1 parent c5782e6
commit 79a0107
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cellfinder_core/tools/IO.py b/cellfinder_core/tools/IO.py
@@ -33,7 +33,12 @@ def read_with_dask(path):
     :return:
     """
 
-    filenames = glob.glob(os.path.join(path, "*.tif"))
+    if path.endswith(".txt"):
+        with open(path, "r") as f:
+            filenames = [line.rstrip() for line in f.readlines()]
+
+    else:
+        filenames = glob.glob(os.path.join(path, "*.tif"))
 
     shape, dtype = get_tiff_meta(filenames[0])
     lazy_arrays = [lazy_imread(fn) for fn in get_sorted_file_paths(filenames)]