python-3.x - 根据颜色分割图像

Question

我在应用 k-means 和 clusters = 3 后获得了一个图像。现在我想根据 k-means 后获得的颜色获得 3 个单独的图像。例如，考虑附加的图像。现在我需要一张图片，它只包含蓝色方 block 。一个有字母 v，一个只有背景有没有可能使用 OpenCV 和 python 来做到这一点。

Answer 1

最通用和最简单的方法是为每个区域使用三种独特的灰色。 (虽然我可以在上图中找到超过三个灰度级，可能是由于 imgur 压缩导致的变化。虽然，在一天结束时，k-means 应该正好给出三个 BGR 值)

gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
unique = np.unique(gray)
c1, c2, c3 = unique[0], unique[1], unique[2]

mask1 = np.zeros_like(gray)
mask1[gray == c1] = 255

mask2 = np.zeros_like(gray)
mask2[gray == c2] = 255

mask3 = np.zeros_like(gray)
mask3[mask3 == c3] = 255