extract colors from 2d array; GPU seeded with kmc2; sorted according to number of samples

get_colors.py

import kmc2
from sklearn.cluster import MiniBatchKMeans
from skimge.color import lab2rgb, rgb2lab

def get_colors(array, clusters=3):
    labs = rgb2lab([array]).squeeze()
    seeding = kmc2.kmc2(labs, clusters)
    kmeans = MiniBatchKMeans(n_clusters = clusters, init=seeding)
    kmeans.fit(np.squeeze(labs))
    LABELS = kmeans.labels_
    colors = kmeans.cluster_centers_
    colors = colors[np.argsort(np.unique(LABELS, return_counts=True)[1])[::-1]]
    return lab2rgb([colors]).squeeze()*255