Question

用python写识别图片主要颜色的程序

需求：
现在C盘底下有一个pic的文件夹，里面有很多图片
需要写一个程序，分析每一张图片的主要颜色（选出一个最能代表整个图片的颜色，算法自定，随便百度看看吧）
结果就是每张图片得出一个rgb值，写在一个记事本里面。。。

Answer 1

# -*- coding: utf-8 -*-

import colorsys
 
def get_dominant_color(image):
    
    #颜色模式转换，以便输出rgb颜色值
    image = image.convert('RGBA')
    
    #生成缩略图，减少计算量，减小cpu压力
    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]):
        # 跳过纯黑色
        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)
        
        # 忽略高亮色
        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
    
if __name__=="__main__":
    from PIL import Image
    import os
    
    path = r'.\\pics\\'
    fp = open('file_color.txt','w')
    for filename in os.listdir(path):
        print path+filename
        try:
            color =  get_dominant_color(Image.open(path+filename))
            fp.write('The color of '+filename+' is '+str(color)+'\n')
        except:
            print "This file format is not support"
    fp.close()

pics文件夹和python程序在一个目录下，产生的文件名file_color.txt也在这个目录下。

看看能否帮到你