如何在python中使用openCV的连接组件和stats？

Question

我正在寻找一个如何在python中使用OpenCV的ConnectedComponentsWithStats()函数的示例,请注意,这仅适用于OpenCV 3或更高版本.官方文档仅显示了C++的API,即使在为python编译时该函数存在.我无法在网上找到它.

Answer 1

该功能的工作原理如下:

# Import the cv2 library
import cv2
# Read the image you want connected components of
src = cv2.imread('/directorypath/image.bmp')
# Threshold it so it becomes binary
ret, thresh = cv2.threshold(src,0,255,cv2.THRESH_BINARY+cv2.THRESH_OTSU)
# You need to choose 4 or 8 for connectivity type
connectivity = 4  
# Perform the operation
output = cv2.connectedComponentsWithStats(thresh, connectivity, cv2.CV_32S)
# Get the results
# The first cell is the number of labels
num_labels = output[0]
# The second cell is the label matrix
labels = output[1]
# The third cell is the stat matrix
stats = output[2]
# The fourth cell is the centroid matrix
centroids = output[3]

标签是输入图像大小的矩阵,其中每个元素的值等于其标签.

统计数据是函数计算的统计数据的矩阵.它的长度等于标签数量,宽度等于统计数量.它可以与OpenCV文档一起使用:

每个标签的统计输出,包括背景标签,请参阅下面的可用统计信息.统计数据通过统计数据[label,COLUMN]访问 ,其中可用列定义如下.

• cv2.CC_STAT_LEFT最左边的(x)坐标,它是水平方向上包含框的包含性开始.
• cv2.CC_STAT_TOP最上面的(y)坐标,它是垂直方向上边界框的包含性开始.
• cv2.CC_STAT_WIDTH边界框的水平大小
• cv2.CC_STAT_HEIGHT边界框的垂直大小
• cv2.CC_STAT_AREA连接组件的总面积(以像素为单位)

质心是一个矩阵,每个质心的x和y位置.该矩阵中的行对应于标签号.

Answer 2

我来过这里几次来记住它是如何工作的，每次我都必须将上面的代码简化为：

_, thresh = cv2.threshold(src,0,255,cv2.THRESH_BINARY+cv2.THRESH_OTSU)
connectivity = 4  # You need to choose 4 or 8 for connectivity type
num_labels, labels, stats, centroids = cv2.connectedComponentsWithStats(thresh , connectivity , cv2.CV_32S)

希望对大家有用:)

Answer 3

补充一下Zack Knopp，如果您使用的是灰度图像，则可以简单地使用：

import cv2
import numpy as np

src = cv2.imread("path\\to\\image.png", 0)
binary_map = (src > 0).astype(np.uint8)
connectivity = 4 # or whatever you prefer

output = cv2.connectedComponentsWithStats(binary_map, connectivity, cv2.CV_32S)

当我尝试Zack Knopp在灰度图像上使用答案时，它不起作用，这就是我的解决方案。