连接未知数量的蒙版

Question

我有一个形状为(height, width, num_of_instances)（例如(2448, 2048, 233)）的布尔值数组。我想返回一个形状为(height, width, 1)的数组，如果在相同位置的233个蒙版中的任何值均为True，则每个值均为True。我并不总是知道实例的数量以及高度和宽度。目前，我正在遍历每个像素：

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

这就像人们期望的那样非常慢。那么，什么是快速而Python化的方法呢？

Answer 1

您尝试过这个吗？

resulting_mask = np.any(masks,axis=2)