搜索二维数组内的子数组（图像识别）[关闭]

本质上，我有一个 numpy 图像数组，我试图查找它是否包含特定 RGB 像素值的 2x2 块。因此，例如，如果我的（简化的）图像数组类似于：

A B C D E F

G H I J K L

M N O P Q R

S T U V W X

我正在尝试检查它是否包含，例如：

J K

P Q

我对 numpy 还很陌生，所以我希望得到任何帮助，谢谢。

这个解决方案怎么样：

1）找出小数组右上左手元素在大数组中的所有位置。

2) 检查大数组中每个给定元素对应的切片是否与小数组完全相同。

假设切片的左上角元素是5，我们会在大数组中找到5的位置，然后检查大数组中5左下角的切片是否与小数组相同。

import numpy as np

a = np.array([[0,1,5,6,7],
              [0,4,5,6,8],
              [2,3,5,7,9]])

b = np.array([[5,6],
              [5,7]])

b2 = np.array([[6,7],
               [6,8],
               [7,9]])

def check(a, b, upper_left):
    ul_row = upper_left[0]
    ul_col = upper_left[1]
    b_rows, b_cols = b.shape
    a_slice = a[ul_row : ul_row + b_rows, :][:, ul_col : ul_col + b_cols]
    if a_slice.shape != b.shape:
        return False
    return (a_slice == b).all()

def find_slice(big_array, small_array):
    upper_left = np.argwhere(big_array == small_array[0,0])
    for ul in upper_left:
        if check(big_array, small_array, ul):
            return True
    else:
        return False

结果：

>>> find_slice(a, b)
True
>>> find_slice(a, b2)
True
>>> find_slice(a, np.array([[5,6], [5,8]]))
False