Make local_correlations in cnmf compatible with 3D data

nel-lab · Jun 22, 2024 · 06b8070 · 06b8070
1 parent 6ac54b2
commit 06b8070
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mesmerize_core/algorithms/cnmf.py b/mesmerize_core/algorithms/cnmf.py
@@ -99,7 +99,7 @@ def run_algo(batch_path, uuid, data_path: str = None):
 
         cnm.save(str(output_path))
 
-        Cn = cm.local_correlations(images.transpose(1, 2, 0))
+        Cn = cm.local_correlations(images, swap_dim=False)
         Cn[np.isnan(Cn)] = 0
 
         corr_img_path = output_dir.joinpath(f"{uuid}_cn.npy").resolve()