sift_svm.py 文件源码

def create_histogram(image, hist_size=2048, codebook=[], detectAndCompute=SIFT_create().detectAndCompute):
    histogram = np.zeros(hist_size)
    descriptors = detectAndCompute(image, window_size=None)
    tree = spatial.KDTree(codebook)

    for i in xrange(len(descriptors)):
        histogram[tree.query(descriptors[i])[1]] += 1

    return normalize(histogram[:, np.newaxis], axis=0).ravel()