fixing bug with multi-plane nonrigid ops

MouseLand · Feb 9, 2024 · 02fe382 · 02fe382
1 parent c6e2915
commit 02fe382
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/suite2p/registration/register.py b/suite2p/registration/register.py
@@ -171,7 +171,7 @@ def compute_reference_masks(refImg, ops=default_ops()):
     if isinstance(refImg, list):
         refAndMasks_all = []
         for rimg in refImg:
-            refAndMasks = compute_reference_masks(rimg)
+            refAndMasks = compute_reference_masks(rimg, ops=ops)
             refAndMasks_all.append(refAndMasks)
         return refAndMasks_all
     else:
@@ -227,7 +227,6 @@ def register_frames(refAndMasks, frames, rmin=-np.inf, rmax=np.inf, bidiphase=0,
 
 
     """
-
     if nZ > 1:
         cmax_best = -np.inf * np.ones(len(frames), "float32")
         cmax_all = -np.inf * np.ones((len(frames), nZ), "float32")
@@ -252,7 +251,6 @@ def register_frames(refAndMasks, frames, rmin=-np.inf, rmax=np.inf, bidiphase=0,
                 outputs = register_frames(refAndMasks[z], frames[[i]], rmin=rmin[z],
                                           rmax=rmax[z], bidiphase=bidiphase, ops=ops,
                                           nZ=1)
-
                 if i == 0:
                     outputs_best = []
                     for output in outputs[:-1]:
@@ -437,7 +435,6 @@ def compute_reference_and_register_frames(f_align_in, f_align_out=None, refImg=N
         bidiphase = 0
 
     refAndMasks = compute_reference_masks(refImg, ops)
-
     ### ------------- register frames to reference image ------------ ###
 
     mean_img = np.zeros((Ly, Lx), "float32")
@@ -642,7 +639,6 @@ def registration_wrapper(f_reg, f_raw=None, f_reg_chan2=None, f_raw_chan2=None,
     else:
         nchannels = 1
 
-
     outputs = compute_reference_and_register_frames(f_align_in, f_align_out=f_align_out,
                                                     refImg=refImg, ops=ops)
     refImg, rmin, rmax, mean_img, rigid_offsets, nonrigid_offsets, zest = outputs