Segmentación por K-means

Para usar el algoritmo de K-means necesitaremos la librería scikit-learn. En este caso, dado una imagen en escala de grises de NxM, podemos invocar el algoritmo de agrupación así:

from sklearn.cluster import KMeans


img_flat = imagen.reshape(N*M,1)
model_kmeans = KMeans(n_clusters=4)
segmented_image = model_kmeans.fit(img_flat).reshape(N,M,1)

En este caso, usamos 4 grupos (n_clusters). Note que la imagen resultante solo indica que píxeles pertenecen a uno de los grupos indicados.