Skip to content

Instantly share code, notes, and snippets.

Last active Jun 23, 2018
What would you like to do?
import os
import shutil
import numpy as np
from PIL import Image
from skimage import data
from sklearn.cluster import KMeans
cluster_count = 4
feature = np.array([data.imread("./icons/" + path) for path in os.listdir('./icons')])
feature = feature.reshape(len(feature), -1).astype(np.float64)
print('data reshape completed')
model = KMeans(n_clusters=cluster_count).fit(feature)
print('clustering completed')
labels = model.labels_
for label, path in zip(labels, os.listdir('./icons')):
dist_dir_name = "icon_groups"
dirpath = dist_dir_name + "/" + str(label)
if not os.path.isdir(dirpath):
shutil.copyfile("./icons/" + path, dist_dir_name + "/" + str(label) + "/" + path)
print(label, path)
print('all completed')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment