tcor_util.py 文件源码

def mclass(cls1,cls2,cls3,nmax):
    num=imax*jmax
    cls1x=np.float32(cls1.reshape(num))
    cls2x=np.float32(cls2.reshape(num))
    cls3x=np.float32(cls3.reshape(num))
    data=np.array([[cls1x],[cls2x],[cls3x]])
    data=data.reshape(3,num)
    data=np.transpose(data)
    datax=data[::100,:]
    t=cv2.getTickCount()
    codebook, destortion = scipy.cluster.vq.kmeans(datax, nmax, iter=10, thresh=1e-05)
    print (cv2.getTickCount()-t)/cv2.getTickFrequency()
    t=cv2.getTickCount()
    code, dist = scipy.cluster.vq.vq(data, codebook)
    print (cv2.getTickCount()-t)/cv2.getTickFrequency()
    return code.reshape(jmax,imax)