python - 如何使用 Numpy 查找两个图像之间的相关性

Question

这是受此启发的question .

我正在尝试使用 Numpy 查找两个灰度图像之间的相关性。使用 SciPy correlate2d我们可以找到这个。我找到了 Numpy 的 corrcoef但与 correlate2d 相比，结果有所不同。因此这个问题 - Numpy 中是否有等效的 correlate2d ？

Answer 1

据我所知，这会产生与 scipy.correlate2d() 相同的结果，其中 img1 和 img2 是二维数组表示灰度(即单 channel )图像:

import numpy as np

pad = np.max(img1.shape) // 2
fft1 = np.fft.fft2(np.pad(img1, pad))
fft2 = np.fft.fft2(np.pad(img2, pad))
prod = fft1 * fft2.conj()
result_full = np.fft.fftshift(np.fft.ifft2(prod))
corr = result_full.real[1+pad:-pad+1, 1+pad:-pad+1]

单像素裁剪调整不是很优雅，但这就是适合您的 FFT:繁琐。

我只是想说 scipy 非常好用，我强烈推荐它。话虽如此，对于我尝试过的单个案例来说，这种方法似乎确实要快得多。