在较大的数组中查找较小的数组

Question

我试图在一张大图中找到一张小图。我已经用PIL完成了，但我正在寻找加速的方法，所以我目前正在寻找一些numpy解决方案。到目前为止，我已经做到： 将两个RGB图像加载到np.0阵列

rows, cols = np.where(np.all(haystack == needle[0,0], axis=-1))
for row, col in zip(rows, cols):
   if row+h > H or col+w > W: continue
   local_haystack = haystack[row:row+h, col:col+w].copy()

这就是我被卡住的地方，我试过np.count\u nonzero，np.sum等等，但问题似乎是，它对任何颜色的计数都像是一个扁平数组而不是元组。所以（255，255，255）不是数1，而是数3，这让我错过了针