如何在np数组中有效地求和出现的次数？遮罩区域RCNN

Question

我正在和图书馆MaskRCNN合作，我想得到每个面具的面积。 所有掩码都是一个数组（W x H），其值为False或True。所有的真代表一个面具，所有的假代表一个背景

#mask 0
r['masks'][:,:,0] 
#output-->
   array([[False, False, False, ..., False, False, False],
   [False, False, False, ..., False, False, False],
   [False, False, False, ..., False, False, False],
   ...,
   [False, False, False, ..., False, False, False],
   [False, False, False, ..., False, False, False],
   [False, False, False, ..., False, False, False]])

如何遍历所有列和行中的所有真实事件并求和

我想这就是掩模的像素面积，我可以插值真实面积

Answer 1

基于这个previous question，您首先需要展平numpy数组，从而应用相同的思想，即

flat_r = numpy.flatten(r['masks'][:,:,0])

unique, counts = numpy.unique(flat_r, return_counts=True)

print(dict(zip(unique, counts))) 
# {False: 7, True: 4}

Answer 2

克劳迪奥的回答很好，但我也找到了一个简单的方法

true_occurrences= np.sum(r['masks'][:,:,0])