mirror of
https://github.com/KrisKennaway/ii-pix.git
synced 2024-06-03 05:29:32 +00:00
Tidy
This commit is contained in:
parent
1ce5c25764
commit
6988b19b43
|
@ -83,13 +83,10 @@ class ClusterPalette:
|
||||||
|
|
||||||
clusters = cluster.MiniBatchKMeans(n_clusters=16, max_iter=10000)
|
clusters = cluster.MiniBatchKMeans(n_clusters=16, max_iter=10000)
|
||||||
clusters.fit_predict(self._colours_cam)
|
clusters.fit_predict(self._colours_cam)
|
||||||
num_colours = clusters.n_clusters
|
|
||||||
|
|
||||||
labels = clusters.labels_
|
|
||||||
|
|
||||||
# Dict of {palette idx : frequency count}
|
# Dict of {palette idx : frequency count}
|
||||||
palette_freq = {idx: 0 for idx in range(16)}
|
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
|
palette_freq[idx] = freq
|
||||||
frequency_order = [
|
frequency_order = [
|
||||||
k for k, v in sorted(
|
k for k, v in sorted(
|
||||||
|
|
Loading…
Reference in New Issue
Block a user