使用VGG_Face模型进行人脸识别时出现的bug: KeyError: 'fc7'

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)