python - opencv python connectedComponents 选择每个标签的组件

Question

我想选择此图像的每个组件：

在实践中，每个三角形都有其标签。我不知道怎么做。我有这个代码：

#!/usr/bin/python
import cv2
import numpy as np
img = cv2.imread('invMehs.png', -1)
imGray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
ret, imBw = cv2.threshold(imGray, 250, 255, cv2.THRESH_BINARY)

invBwMesh = cv2.bitwise_not(imBw)
Mask = np.ones(imBw.shape, dtype="uint8") * 255

connectivity = 4
output = cv2.connectedComponentsWithStats(imBw, connectivity, cv2.CV_32S)
num_labels = output[0]
labels = output[1]
stats = output[2]
centroids = output[3]
labels = labels + 1

b = ( labels == 1)
cv2.imwrite('tst.jpg',labels[b])

但是图像是完全黑色的：S 非常感谢。

Answer 1

您要保存的图像（标签 [b]）仅包含细线（灰度 1）。使用 JPEG 格式保存图像时，压缩算法会对其进行平滑处理，但由于它们与背景只有 1 个灰度级，因此会被删除。这就是为什么你会得到一个黑色的图像

以 PNG 格式保存不会更改图像标签。

为了保留每个连接组件的所有标签（背景为 0），要编写的代码应该是：

cv2.imwrite('labels.png',output[1])