np.sum 不返回总计数

Question

我正在尝试此代码，但它不返回零[x][y] 的总计数，在这种情况下它应该返回 5 但它全部显示 255 五次。
此代码用于连接组件，零是白色像素的一个组件

预期输出应该是：5
但我得到的是：
255
255
255
255
255

for (x, y) in labels:
        component = uf.find(labels[(x, y)])
        # Update the labels with correct information
        labels[(x, y)] = component
        path='imagesNew/13/'
        if labels[(x, y)]==0:
            Zero[y][x]=int(255)
            Z=Zero[y][x]
            print Z
            if Z==5:
              print Zero[y][x]
              Zeroth = Image.fromarray(Zero)
              Zeroth.save(path+'Zero'+'.png','png')

Answer 1

我相信这就是您所追求的。此代码将遍历标记图像中的每个标签，创建一个布尔掩码，显示每个标签在图像中的位置，并将该掩码求和以告诉您每个标签的总数。

import numpy as np
labeled_img = np.array([
    [1, 1, 0, 0, 2],
    [1, 0, 0, 2, 2],
    [1, 0, 2, 2, 2],
    [1, 1, 1, 2, 2],
    [1, 1, 1, 2, 2]])

for label in np.unique(labeled_img):
    print('Quantity of', label, 'labels:', np.sum(labeled_img == label))

0 个标签的数量：5
1 个标签的数量：10
2 个标签的数量：10

请注意，如果您使用 OpenCV 查找连接的组件，connectedComponents() 的第一个返回值是标签的数量，因此您可以简单地循环遍历 for label in range(n_labels) 而不是 np.unique()，这样会快一点.