mp_fourier: call python version for tiny images

dstndstn · Dec 11, 2017 · 30fb8aa · 30fb8aa
1 parent 7d57e76
commit 30fb8aa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tractor/psf.py b/tractor/psf.py
@@ -32,7 +32,8 @@ def lanczos_shift_image(img, dx, dy, inplace=False, force_python=False):
     Ly /= Ly.sum()
 
     #print('mp_fourier:', mp_fourier)
-    if mp_fourier is None or force_python:
+    H,W = img.shape
+    if mp_fourier is None or force_python or W <= 8 or H <= 8:
         sx     = correlate1d(img, Lx, axis=1, mode='constant')
         outimg = correlate1d(sx,  Ly, axis=0, mode='constant')
     else: