Extract list of color from an Image using python

In this tutorial I will show you how to extract list of color from an image using python program.

PYTHON CODE:

import numpy as np

import matplotlib.pyplot as plt

from sklearn.cluster import KMeans

from PIL import Image



image = Image.open('mobile.jpg')

image_np = np.array(image)



w, h, d = image_np.shape



pixels = image_np.reshape(-1, 3)



n_colors = 10



kmeans = KMeans(n_clusters=n_colors, random_state=42, n_init=10)

kmeans.fit(pixels)



colors = kmeans.cluster_centers_.astype(np.uint8)



palette = np.reshape(colors, (1, -1, 3))



plt.imshow(palette)

plt.axis('off')

plt.show()

Install necessary package:

pip install pillow numpy matplotlib scikit-learn

OUTPUT:

VIDEO GUIDE: