test.py 文件源码

def compar_pic(path1,path2):
    global net
    #??????
    X=read_image(path1)
    test_num=np.shape(X)[0]
    #X  ?? ?????
    out = net.forward_all(blobs=['pool5'],data = X)
    # print out.keys()

    feature1 = np.float64(out["pool5"])

    feature1=np.reshape(feature1,(test_num,25088))
    #??????
    X=read_image(path2)
    #X  ?? ?????
    out = net.forward_all(blobs=['pool5'],data=X)
    #fc7??????,??????
    feature2 = np.float64(out['pool5'])
    feature2=np.reshape(feature2,(test_num,25088))
    #????????cos?,??????????
    predicts=pw.cosine_similarity(feature1, feature2)
    return  predicts