【OpenCV】47 二值图像连通组件状态统计

47 二值图像连通组件状态统计

代码

import cv2 as cv
import numpy as np

def connected_components_stats_demo(src):
    src = cv.GaussianBlur(src, (3, 3), 0)
    gray = cv.cvtColor(src, cv.COLOR_BGR2GRAY)
    ret, binary = cv.threshold(gray, 0, 255, cv.THRESH_BINARY | cv.THRESH_OTSU)
    cv.imshow("binary", binary)

    num_labels, labels, stats, centers = cv.connectedComponentsWithStats(binary, connectivity=8, ltype=cv.CV_32S)
    colors = []
    for i in range(num_labels):
        b = np.random.randint(0, 256)
        g = np.random.randint(0, 256)
        r = np.random.randint(0, 256)
        colors.append((b, g, r))

    colors[0] = (0, 0, 0)
    image = np.copy(src)
    for t in range(1, num_labels, 1):
        x, y, w, h, area = stats[t]
        cx, cy = centers[t]
        cv.circle(image, (np.int32(cx), np.int32(cy)), 2, (0, 255, 0), 2, 8, 0)
        cv.rectangle(image, (x, y), (x+w, y+h), colors[t], 1, 8, 0)
        cv.putText(image, "num:%s"%t, (x, y), cv.FONT_HERSHEY_SIMPLEX, .5, (0, 0, 255), 1);
        print("label index %d, area of the label : %d"%(t, area))

    cv.imshow("colored labels", image)
    # cv.imwrite("D:/labels.png", image)
    print("total rice : ", num_labels - 1)

input = cv.imread("../images/rice.png")
connected_components_stats_demo(input)
cv.waitKey(0)
cv.destroyAllWindows()

实验结果

解释

OpenCV中会输出连通组件统计信息的相关API，
int cv::connectedComponentsWithStats(
InputArray image,
OutputArray labels,
OutputArray stats,
OutputArray centroids,
int connectivity,
int ltype,
int ccltype
)

retval, labels, stats, centroids	=	cv.connectedComponentsWithStats(image[, labels[, stats[, centroids[, connectivity[, ltype]]]]])

• retval：返回的label数目的数值
• image：输入二值图像，黑色背景
• labels：输出的标记图像，背景index=0
• stats：统计信息，包括每个组件的位置、宽、高与面积
• centroids：每个组件的中心位置坐标cx, cy connectivity,// 寻找连通组件算法的连通域，默认是8连通
• ltype：输出的labels的Mat类型CV_32S
• ccltype：/连通组件算法

相关的统计信息包括在输出stats的对象中，每个连通组件有一个这样的输出结构体。
CC_STAT_LEFT
Python: cv.CC_STAT_LEFT
连通组件外接矩形左上角坐标的X位置信息

CC_STAT_TOP
Python: cv.CC_STAT_TOP
连通组件外接左上角坐标的Y位置信息

CC_STAT_WIDTH
Python: cv.CC_STAT_WIDTH
连通组件外接矩形宽度

CC_STAT_HEIGHT
Python: cv.CC_STAT_HEIGHT
连通组件外接矩形高度

CC_STAT_AREA
Python: cv.CC_STAT_AREA
连通组件的面积大小，基于像素多少统计。

Centroids输出的是每个连通组件的中心位置坐标(x, y)

---

所有内容均来源于贾志刚老师的知识星球——OpenCV研习社，本文为个人整理学习，已获得贾老师授权，有兴趣、有能力的可以加入贾老师的知识星球进行深入学习。