Opencv快速获取连通域

对于ndarray数据中的连通域查找，opencv提供了接口，非常方便。

import cv2
import numpy as np

img = np.array([
    [0, 255, 255, 0, 0, 0, 255, 255,],
    [0, 0, 255, 0, 255, 255, 255, 0],
    [0, 0, 0,  0, 255, 255, 0, 255],
    [255, 255, 0, 0, 0, 0, 0, 0],
    [255, 255, 0, 0, 0, 0, 0, 0],
    [255, 255, 0, 0, 0, 0, 0, 0]
], dtype=np.uint8)

num, labels = cv2.connectedComponents(img)
labels_dict = {i:[] for i in range(1, num+1)}
height, width = img.shape
for h in range(height):
    for w in range(width):
        if labels[h][w] in labels_dict:
            labels_dict[labels[h][w]].append([h,w])

cv2.connectedComponents()函数返回查找到的连通域个数和对应的label。

上面代码返回连通域个数为4（包含值为0区域，可通过lables过滤）, labels结果如图所示：