如何在python中对二进制图像进行组件标记

Question

我试图通过简单数组的轮廓跟踪来应用组件标签作为示例。

arr = np.array([
                [1,0,1,0,0,0,0],
                [1,1,1,0,0,0,0],
                [0,1,1,0,0,0,1],
                [0,1,1,0,0,1,1],
                [0,0,0,0,1,1,1],
                [0,0,0,1,1,1,1],
                [0,0,0,1,1,1,1],
                ])

这表示一个二值图像，0 表示空白，1 表示形状。

我试图得到的结果是分别标记这两个多边形，并通过 matplotlib 在图表中以不同的颜色显示每个多边形（证明多边形中的每个点都已标记到相应的区域。

我认为skimage.measure.regionprops、skimage.measure.label 和skimage.measure.find_contours 的组合可以解决问题，但我找不到任何我想要解决的示例。

我花了几个小时试图理解文档并搜索以前的帖子，但现在陷入了死胡同。 This post here似乎与我的问题类似，尽管我希望能够标记形状内部的每个像素，而不仅仅是周长。

非常感谢任何对我应该做的事情的帮助或解释。谢谢

Answer 1

你只需要使用skimage.measure.label:

import numpy as np
from skimage.measure import label
from skimage import io

arr = np.array([[1,0,1,0,0,0,0],
                [1,1,1,0,0,0,0],
                [0,1,1,0,0,0,1],
                [0,1,1,0,0,1,1],
                [0,0,0,0,1,1,1],
                [0,0,0,1,1,1,1],
                [0,0,0,1,1,1,1]])

img = label(arr)
io.imshow(img)

In [12]: img
Out[12]: 
array([[1, 0, 1, 0, 0, 0, 0],
       [1, 1, 1, 0, 0, 0, 0],
       [0, 1, 1, 0, 0, 0, 2],
       [0, 1, 1, 0, 0, 2, 2],
       [0, 0, 0, 0, 2, 2, 2],
       [0, 0, 0, 2, 2, 2, 2],
       [0, 0, 0, 2, 2, 2, 2]], dtype=int64)