Skip to content

Commit

Permalink
Tiny ufunc fix in motion_estimation
Browse files Browse the repository at this point in the history
  • Loading branch information
alejoe91 committed Feb 6, 2024
1 parent 5f1b6cb commit 197b73a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/spikeinterface/sortingcomponents/motion_estimation.py
Original file line number Diff line number Diff line change
Expand Up @@ -322,7 +322,7 @@ def run(

if histogram_depth_smooth_um is not None:
bins = np.arange(motion_histogram.shape[1]) * bin_um
bins -= np.mean(bins)
bins = bins - np.mean(bins)
smooth_kernel = np.exp(-(bins**2) / (2 * histogram_depth_smooth_um**2))
smooth_kernel /= np.sum(smooth_kernel)
motion_histogram = scipy.signal.fftconvolve(motion_histogram, smooth_kernel[None, :], mode="same", axes=1)
Expand Down

0 comments on commit 197b73a

Please sign in to comment.