11.20聚类色彩分析

一、导入包

from flask import Flask,render_template
import numpy as np
from PIL import Image
from scipy.cluster.vq import vq, kmeans, whiten
import ImageColor

二、定义函数，用缩略图聚类

app = Flask(__name__)

#用缩略图聚类
def colorz(filename, n=3):
    img=Image.open(filename)
    img=img.rotate(-90)
    img.thumbnail((200,200))
    w,h=img.size
    print(w,h)
    print('w*h=',w*h)
    points=[]
    for count,color in img.getcolors(w*h):
        points.append(color)
    return points

三、对色彩聚类

#对色彩聚类
def kmeansColor(img,n):
    points=colorz(img,3)
    fe = np.array(points,dtype=float)
    codebook, distortion = kmeans(fe,n)
    centers=np.array(codebook,dtype=int)
    return centers

四、显示在网页端

@app.route("/")
def index():
    #genFrame()
    pic = "static/hist/image"
    framecount=249
    imgcolors &#