def compar_pic(path1,path2):
global net
#加载验证图片
X=read_image(path1)
test_num=np.shape(X)[0]
#X 作为 模型的输入
#out = net.forward_all(data = X) 这里改为下行,增加blobs=['fc7'],可避免报错
out = net.forward_all(blobs=['fc7'],data = X)
#fc7是模型的输出,也就是特征值
feature1 = np.float64(out['fc7'])
feature1=np.reshape(feature1,(test_num,4096))
#np.savetxt('feature1.txt', feature1, delimiter=',')
#加载注册图片
X=read_image(path2)
#X 作为 模型的输入
#这里也同样改动
out = net.forward_all(blobs=['fc7'],data = X)
#fc7是模型的输出,也就是特征值
feature2 = np.float64(out['fc7'])
feature2=np.reshape(feature2,(test_num,4096))
#np.savetxt('feature2.txt', feature2, delimiter=',')
#求两个特征向量的cos值,并作为是否相似的依据
predicts=pw.cosine_similarity(feature1, feature2)
return predicts
原本是这样写的:out = net.forward_all(data = X)
然后报错KeyError: 'fc7'
这里改动,增加blobs=['fc7'],可避免报错
out = net.forward_all(blobs=['fc7'],data = X)