Python gets the main color of the image and contrasts it with the color library by PIL

This example describes how Python obtains the main color of a picture and contrasts it with a color library through PIL. Share to everyone for your reference. The specific analysis is as follows:

This code is mainly used to extract its main color from the image, similar to Goolge and Baidu image search can be specified by color search, so we need to extract the main color of each picture, and then divide the color into its closest color segment, then you can search by color.

In the use of Google or Baidu search will find a picture color options, it feels very interesting, some people may think this is definitely artificial to divide, hehe, there is this possibility, but estimated that people will be exhausted, open a joke, of course, is through the machine recognition, a huge amount of pictures only machine recognition to do.

Is it possible to do this with python? The answer is: can

With the powerful image processing capabilities of the Python PIL module, the following code is available:
Copy the Code code as follows:

Import Colorsys
def get_dominant_color (image):
#颜色模式转换 in order to output RGB color values
Image = Image.convert (' RGBA ')
#生成缩略图, reduce computational load and reduce CPU pressure
Image.thumbnail ((200, 200))
Max_score = None
Dominant_color = None
For Count, (R, G, B, a) in Image.getcolors (image.size[0] * image.size[1]):
# Skip Pure Black
If a = = 0:
Continue
saturation = COLORSYS.RGB_TO_HSV (r/255.0, g/255.0, b/255.0) [1]
y = min (abs (R * 2104 + G * 4130 + b * 802 + 4096 + 131072) >> 13, 235)
y = (y-16.0)/(235-16)
# Ignore High colors
If y > 0.9:
Continue
# Calculate The score, preferring highly saturated colors.
# ADD 0.1 to the saturation so we don ' t completely ignore grayscale
# Colors by multiplying the count by zero, but still give them a low
# weight.
Score = (saturation + 0.1) * Count
If score > Max_score:
Max_score = Score
Dominant_color = (r, G, b)
Return Dominant_color

How to use:

From PIL import ImagePrint get_dominant_color (Image.open (' logo.jpg '))

This will return an RGB color, but this value is very accurate range, then how do we achieve the color gamut of Baidu image??

In fact, the method is very simple, r/g/b are 0-255 of the value, we just have to divide the three values into equal intervals, and then combine, take the approximate value. For example: Divided into 0-127, and 128-255, and then free combination, can appear eight combinations, and then choose a more representative color.

Of course I just give an example, you can also divide the finer, so that the color of the display will be more accurate ~ ~ Everybody try it quickly

Hopefully this article will help you with Python programming.

