在Numpy图像中查找子图像

Question

我有两个从PIL图像转换的Numpy数组（3维uint8）。

我想查找第一张图像是否包含第二张图像，如果是，则找出匹配的第一张图像内左上像素的坐标。

有没有一种方法可以在Numpy中以足够快的方式完成此操作，而不是使用（4！非常慢）纯Python循环？

2D示例：

a = numpy.array([
    [0, 1,  2,  3],
    [4, 5,  6,  7],
    [8, 9, 10, 11]
])
b = numpy.array([
    [2, 3],
    [6, 7]
])

怎么做这样的事情？

position = a.find(b)

position然后将是(0, 2)。

Answer 1

这可以使用SciPy的公司来完成correlate2d，然后用argmax找到互相关的峰值。

这是对数学和思想的更完整说明，并提供了一些示例。

如果您想呆在纯粹的Numpy中，甚至不使用scipy，或者图像很大，则最好使用基于FFT的互相关方法。

编辑：这个问题专门要求一个 纯粹的Numpy解决方案
。但是，如果可以使用OpenCV或其他图像处理工具，则使用其中一种显然会更容易。下面的PiQuer提供了这样的示例，如果可以使用，我建议您使用它。