Объединение неизвестного количества масок - PullRequest
0 голосов
/ 06 мая 2020

У меня есть массив логических значений формы (height, width, num_of_instances) (например, (2448, 2048, 233)). Я хотел бы вернуть массив формы (height, width, 1), где каждое значение истинно, если любое из значений в 233 масках в той же позиции истинно. Я не всегда знаю количество экземпляров, высоту и ширину. В настоящее время я перебираю каждый пиксель:

for j in masks.shape[0]:
   for i in masks.shape[1]:
      if any(masks[j, i, :):
          resulting_mask[j, i] = True

Это очень медленно, как и следовало ожидать. Итак, что было бы быстрым и c * питонским * способом сделать это?

1 Ответ

0 голосов
/ 06 мая 2020

Вы пробовали это?

resulting_mask = np.any(masks,axis=2)[:,:,None]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...