在Numpy图像中查找子图像

import numpy as np

# 假设原始图像为image，子图像为sub_image
# 将图像转换为Numpy数组
image = np.array(image)
sub_image = np.array(sub_image)

# 获取子图像的形状
sub_image_shape = sub_image.shape

# 使用np.where函数查找子图像在原始图像中的位置
rows, cols = np.where((image == sub_image).all(axis=-1))

# 获取子图像在原始图像中的位置
for row, col in zip(rows, cols):
    print(f"子图像在原始图像中的位置为：({row}, {col})")