Как эффективно сгенерировать все срезы numpy массива X, чтобы они имели форму, равную форме numpy массива Y? - PullRequest
Новая
       65

Как эффективно сгенерировать все срезы numpy массива X, чтобы они имели форму, равную форме numpy массива Y?

0 голосов
/ 14 апреля 2020

Допустим, у меня есть два numpy массива X и Y, которые представляют два изображения, где X.shape[0] > Y.shape[0] равно True и X.shape[1] > Y.shape[1] равно True. Я хотел бы сгенерировать все 2D-срезы X так, чтобы каждый срез x удовлетворял этому x.shape == Y.shape.

Кажется, что следующая попытка работает, но очень медленно: 

def window(X, width, height):
    for i in range(X.shape[0]-height):
        for j in range(X.shape[1]-width):
            yield X[i:i+height,j:j+width]

Как заставить этот генератор работать быстрее?

...