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: