utils.py 文件源码

def GetKnnIdx(queryData,baseData,numNN, metric=0):
    if (metric==0):
        objMatcher=cv2.BFMatcher(cv2.NORM_L2)
    elif (metric==1):
        objMatcher=cv2.BFMatcher(cv2.NORM_HAMMING)
    matches=objMatcher.knnMatch(queryData,baseData,k=numNN)
    idxKnn=npy.zeros((queryData.shape[0],numNN), dtype=npy.int32)
    for kk in range(queryData.shape[0]):
        for ll in range(numNN):
            idxKnn[kk][ll]=matches[kk][ll].trainIdx
    return idxKnn