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
image
class 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
,g
andb
.
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)