Include background component indices when removing empty/nan components

flatironinstitute · Nov 14, 2024 · 7377213 · 7377213
1 parent ce2cb17
commit 7377213
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/caiman/source_extraction/cnmf/spatial.py b/caiman/source_extraction/cnmf/spatial.py
@@ -180,7 +180,7 @@ def update_spatial_components(Y, C=None, f=None, A_in=None, sn=None, dims=None,
         A_in = caiman.utils.stats.csc_column_remove(A_in, list(ff))
         C = np.delete(C, list(ff), 0)
         # update indices
-        ind_list = list(range(nr-np.size(ff)))
+        ind_list = list(range(nr+nb-np.size(ff)))
         for i in ff:
             ind_list.insert(i, 0)
         ind_list = np.array(ind_list, dtype=int)