K-means聚类算法之程序实现-3维像素级分割

最近看了k-means算法，网上这个算法很多都是用来分割二维图片的，所以想尝试用这个算法做一下三维图片的分割。
那么我们先来介绍一下二维图片的分割：
原理是这样的，我们把二维图片中的每一个像素点的值作为一个特征值，如果是彩色图片，那么一个像素点就可以由3个特征值组成，因为有三个颜色通道。

代码：

from scipy.cluster.vq import *
from scipy.misc import imresize
from pylab import *
from PIL import Image
import pdb
steps = 40 # image is divided in steps*steps region   

infile = 'E:\dataset\ORI_dataset\ADNI-slice3\cMCI\\278_brain\\46_278_brain.jpg'

im = array(Image.open(infile))

dx = int(im.shape[0] / steps)

dy = int(im.shape[1] / steps)

# compute color features for each region

features = []
#pdb.set_trace()

for x in range(steps):

    for y in range(steps):
        R = mean(im[x * dx:(x + 1) * dx, y * dy:(y + 1) * dy])   #彩色图片进行三次
        #R = mean(im[x * dx:(x + 1) * dx, y * dy:(y + 1) * dy, 0])        
        #G = mean(im[x * dx:(x + 1) * dx, y * dy:(y + 1) * dy, 1])