Как устранить дополнительные измерения в сопоставлении с образцом 3D-массива Numpy? ⇐ Python

[Anonymous]

Этот скрипт Python:

Код: Выделить всё

import numpy as np

img = np.zeros((5, 5, 3), dtype='u1')
a = (1, 2, 3)
b = (3, 2, 1)
img[1:4, 1:4] = a
img[1, 1] = b
pat = img[1:4, 1:4]
win = np.lib.stride_tricks.sliding_window_view(img, window_shape=(3, 3, 3))
matches = np.all(win == pat, axis=(0, 1))
print(matches)

производит:

Код: Выделить всё

[[[[False False False]
[False False False]
[False False False]]

[[False False False]
[False  True False]
[False False False]]

[[False False False]
[False False False]
[False False False]]]]

который имеет избыточные размеры. Верхний левый угол шаблона pat расположен в координатах (1, 1) изображения img. Есть ли способ получить совпадения, равные:

Код: Выделить всё

  [[False False False]
[False  True False]
[False False False]]

чтобы получить координаты узора более напрямую?

Подробнее здесь: https://stackoverflow.com/questions/798 ... n-matching