Detección de color de imagen usando Python

Question

Después de descargar una imagen del sitio, necesito detectar el color de la imagen descargada. He descargado la imagen correctamente pero necesito detectar el color de la imagen correspondiente y guardarla en el nombre del color correspondiente. El código utilizado se proporciona a continuación. Dígame cómo puedo lograrlo desde la posición actual.

imageurl='http://www.example.com/' 
opener1 = urllib2.build_opener() 
page1=opener1.open(imageurl) 
my_picture=page1.read() 
fout = open('images/tony'+image[s], "wb") 
fout.write(my_picture) 
fout.close() 

Answer 1

Use a PIL (Python Image Library) histogram. bucle sobre el histograma y tomar el promedio de color de píxel ponderado por el número de píxeles.

Answer 2

Puede usar el módulo de imagen de la biblioteca PIL para hacer esto. Ver: http://effbot.org/imagingbook/image.htm.

Answer 3

Según lo mencionado por otros, PIL es la biblioteca correcta. Aquí hay una función que abre una imagen y busca el color principal.

def get_main_color(file): 
    img = Image.open(file) 
    colors = img.getcolors(256) #put a higher value if there are many colors in your image 
    max_occurence, most_present = 0, 0 
    try: 
     for c in colors: 
      if c[0] > max_occurence: 
       (max_occurence, most_present) = c 
     return most_present 
    except TypeError: 
     raise Exception("Too many colors in the image") 

espero que ayude

Actualización: pasando de 256 a getcolors está bien para imágenes muy pequeñas, pero no puede trabajar en la mayoría de los casos. Este valor debe aumentarse para imágenes más grandes. por ejemplo, 1024 * 1024 está bien para 400 píxeles * imagen de 300 píxeles.

Answer 4

Debería utilizar el Analizador PIL de la clase ImageFile para leer el archivo desde la URL. Entonces la vida es bastante fácil porque dijiste que toda la imagen es del mismo color. Aquí hay un código que se basa en su código:

import urllib2 
import ImageFile 

image_url = "http://plainview.files.wordpress.com/2009/06/black.jpg" 
opener1 = urllib2.build_opener() 
page1=opener1.open(image_url) 

p = ImageFile.Parser() 

while 1: 
    s = page1.read(1024) 
    if not s: 
     break 
    p.feed(s) 

im = p.close() 
r,g,b = im.getpixel((0,0)) 

fout = open('images/tony'+image[s]+"%d%_d%_d"%(r,g,b), "wb") 
fout.write(my_picture) 
fout.close() 

Esto debería añadir los valores de rojo verde y azul del color del primer píxel de la imagen al final del nombre de la imagen. Probé todo hasta las líneas de búsqueda.

Answer 5

Puede usar el algoritmo K-means para obtener los K colores principales de la imagen. Aquí hay un ejemplo de cómo hacerlo: K-means using OpenCV (Python)