1. Learn
  2. /
  3. Courses
  4. /
  5. Cluster Analysis in Python

Exercise

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.

Instructions

100 XP
  • Import image class of matplotlib.
  • 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 and b.