中国mooc北京理工大学机器学习第一周（三）

     
    def loadData(filePath):
        f = open(filePath,‘rb‘)
        data = []
        img = image.open(f)
        m,n = img.size
        for i in range(m):
            for j in range(n):
                x,y,z = img.getpixel((i,j))  #getpixel
                data.append([x/256.0,y/256.0,z/256.0])
        f.close()
        return np.mat(data),m,n
     
    imgData,row,col = loadData(‘Desktop\\12.jpg‘)#双斜杠
    
    label = KMeans(n_clusters=8).fit_predict(imgData)#进行聚类，找同类
    print(label)
    label = label.reshape([row,col])
    print(label)
    pic_new = image.new("L", (row, col))
    for i in range(row):
        for j in range(col):
            pic_new.putpixel((i,j), int(256/(label[i][j]+1)))#双循环putpixel
    pic_new.save("Desktop\\12-8.jpg", "JPEG")