From ee42c6a9eb01d7eb26980ee612deec69e2ff212c Mon Sep 17 00:00:00 2001
From: Andrey Aristov <aaristov@pasteur.fr>
Date: Sat, 26 Mar 2022 11:56:08 +0100
Subject: [PATCH] detect channel axis

---
 src/napari_segment/_reader.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/napari_segment/_reader.py b/src/napari_segment/_reader.py
index 80560e0..c86f0cd 100644
--- a/src/napari_segment/_reader.py
+++ b/src/napari_segment/_reader.py
@@ -40,11 +40,15 @@ def napari_get_reader(path):
 def read_nd2(path):
     data = nd2.ND2File(path)
     ddata = data.to_dask()
+    try:
+        channel_axis = list(d.sizes.keys()).index('C')
+    except ValueError:
+        channel_axis = None
     return [
         (
             ddata,
             dict(
-                channel_axis=1,
+                channel_axis=channel_axis,
                 name=["BF", "fluo"],
                 colormap=["gray", "green"],
                 contrast_limits=[(8500, 35000), (150, 20000)],
-- 
GitLab