如何使用 scipy.ndimage 找到最大的连接区域？

Question

我有一个二进制图像。二值图像有一些隔离区域，如噪声。我知道预期区域比这些隔离区域大得多。因此，我使用连接组件通过找到最大的连接区域来删除隔离区域。我必须使用 scipy 包。我发现它有一些功能可以做到这一点。然而，我离结果还很远。如何使用这些函数获得可以忽略孤立区域的二值图像？谢谢

from scipy import ndimage
label_im, nb_labels = ndimage.label(binary_img)

# Find the largest connected component
sizes = ndimage.sum(binary_img, label_im, range(nb_labels + 1))
mask_size = sizes < 1000
remove_pixel = mask_size[label_im]
label_im[remove_pixel] = 0
labels = np.unique(label_im)
binary_img= np.searchsorted(labels, label_im)
#Select the biggest connected component
binary_img[binary_img < binary_img.max()]=0
binary_img[binary_img >= binary_img.max()]=1

Answer 1

您有一个良好的开端，ndimage.sum用于查找每个标记区域的大小。

从那里，您可以使用sizes（或从它派生的东西）作为查找表：

from scipy import ndimage
label_im, nb_labels = ndimage.label(binary_img)
sizes = ndimage.sum(binary_img, label_im, range(nb_labels + 1))
mask = sizes > 1000
binary_img = mask[label_im]

这将创建一个查找表mask，true用于与较大区域和false其他地方的标签相对应的索引。使用标记图像对查找表进行索引会产生所需的二值图像。

请注意，这sizes[label_im]是一个图像，其中每个区域都以其大小绘制。也就是说，区域#1 中的每个像素都获得区域#1 的大小值。您可以阈值此图像以删除小区域：

size_img = sizes[label_im]
binary_img = size_img > 1000

（这两行相当于前面代码片段的最后两行。）