From 6988b19b4375796e1b627d4bcd9fd3c78af11771 Mon Sep 17 00:00:00 2001 From: kris Date: Tue, 23 Nov 2021 14:00:57 +0000 Subject: [PATCH] Tidy --- convert.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/convert.py b/convert.py index 35a288f..23101f5 100644 --- a/convert.py +++ b/convert.py @@ -83,13 +83,10 @@ class ClusterPalette: clusters = cluster.MiniBatchKMeans(n_clusters=16, max_iter=10000) clusters.fit_predict(self._colours_cam) - num_colours = clusters.n_clusters - - labels = clusters.labels_ # Dict of {palette idx : frequency count} palette_freq = {idx: 0 for idx in range(16)} - for idx, freq in zip(*np.unique(labels, return_counts=True)): + for idx, freq in zip(*np.unique(clusters.labels_, return_counts=True)): palette_freq[idx] = freq frequency_order = [ k for k, v in sorted(