Fix convolution input to get correlation results.

2026-07-30 12:31:08 +08:00 · 2012-05-08 21:32:07 -04:00
parent 7caf85a5c1
commit 5d79537566
1 changed files with 4 additions and 2 deletions
@@ -101,8 +101,10 @@ def match_template(np.ndarray[float, ndim=2, mode="c"] image,
    inv_area = 1.0 / (template.shape[0] * template.shape[1])

    # when `dtype=float` is used, ascontiguousarray returns ``double``.
-    corr = np.ascontiguousarray(fftconvolve(image, np.fliplr(template),
-                                            mode="valid"), dtype=np.float32)
+    corr = np.ascontiguousarray(fftconvolve(image,
+                                            np.flipud(np.fliplr(template)),
+                                            mode="valid"),
+                                dtype=np.float32)

    cdef int i, j
    cdef float den, window_sqr_sum, window_mean_sqr, window_sum,