Extract RGB values from image
There are broadly three steps to find the dominant colors in an image:
- Extract RGB values into three lists.
- Perform k-means clustering on scaled RGB values.
- Display the colors of cluster centers.
To extract RGB values, we use the imread() function of the image class of matplotlib. Empty lists, r, g and b have been initialized.
For the purpose of finding dominant colors, we will be using the following image.
This is a part of the course
“Cluster Analysis in Python”
Exercise instructions
- Import
imageclass ofmatplotlib. - Read the image using the
imread()function and print the dimensions of the resultant matrix. - Store the values for the three colors from all pixels in lists
r,gandb.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Import image class of matplotlib
____ as img
# Read batman image and print dimensions
batman_image = ____('batman.jpg')
print(____)
# Store RGB values of all pixels in lists r, g and b
for ____:
for temp_r, temp_g, temp_b in ____:
r.append(temp_r)
g.append(temp_g)
b.append(temp_b)This exercise is part of the course
Cluster Analysis in Python
In this course, you will be introduced to unsupervised learning through techniques such as hierarchical and k-means clustering using the SciPy library.
What is DataCamp?
Learn the data skills you need online at your own pace—from non-coding essentials to data science and machine learning.