Как узнать количество кластеров на изображении?

Question

Найдите количество кластеров на этом изображении:

Я пытаюсь найти номер кластера на этом изображении. Я пробовал openCV morphologyEx и разрушать, но не могу получить один пиксель для каждого кластера. Пожалуйста, предложите, какой будет лучший способ подсчитать количество кластеров в изображении, используя openCV, предпочтительно на Python.

- Edit

Я попытался прореживать, разрушать и morphologyEx (закрытие), но не смог объединить кластеры в один пиксель. Ниже приведены некоторые из вещей, которые я попробовал.

kernel = np.ones((2, 2), np.uint8) #[[1,1,1],[1,1,1],[1,1,1]
erosion = cv2.erode(img, kernel, iterations=1)
closing = cv2.morphologyEx(img, cv2.MORPH_CLOSE, kernel)
cv2.imwrite('test1.jpg', erosion)
cv2.imwrite('test2.jpg', closing)

img = cv2.imread(file, 0)
size = np.size(img)
skel = np.zeros(img.shape, np.uint8)

#ret, img = cv2.threshold(img, 127, 255, 0)
element = cv2.getStructuringElement(cv2.MORPH_CROSS, (3, 3))
done = False

while (not done):
    eroded = cv2.erode(img, element)
    temp = cv2.dilate(eroded, element)
    temp = cv2.subtract(img, temp)
    skel = cv2.bitwise_or(skel, temp)
    img = eroded.copy()

    zeros = size - cv2.countNonZero(img)
    if zeros == size:
        done = True

cv2.imwrite('thinning.jpg', skel)

Answer 1

Решение так же просто, как это. Вы должны найти количество контуров изображения и сосчитать их. Для этого вы можете использовать метод cv2.findContours со следующими параметрами. Для получения более подробной информации о cv2.findContours, пожалуйста, проверьте документацию .

import cv2
img = cv2.imread('test.jpg', 0)
cv2.threshold(img,0,255,cv2.THRESH_BINARY+cv2.THRESH_OTSU,img)

image, contours, hier = cv2.findContours(img, cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE)
count = len(contours)
print(count)

выход:

973

Answer 2

Как это?

import numpy as np
import cv2

img = cv2.imread('points.jpg')
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
ret, thresh = cv2.threshold(gray,0,255,cv2.THRESH_BINARY+cv2.THRESH_OTSU)

n_labels, labels, stats, centroids = cv2.connectedComponentsWithStats(thresh)

print(n_labels)

size_thresh = 1
for i in range(1, n_labels):
    if stats[i, cv2.CC_STAT_AREA] >= size_thresh:
        #print(stats[i, cv2.CC_STAT_AREA])
        x = stats[i, cv2.CC_STAT_LEFT]
        y = stats[i, cv2.CC_STAT_TOP]
        w = stats[i, cv2.CC_STAT_WIDTH]
        h = stats[i, cv2.CC_STAT_HEIGHT]
        cv2.rectangle(img, (x, y), (x+w, y+h), (0, 255, 0), thickness=1)

cv2.imwrite("out.jpg", img)

номер кластера: 974
out.jpg: