NotFoundError (see above for tra

2019-05-10  本文已影响0人  Xfyyzy

在restore已训练好模型的时候,验证第二张图片时报以下错误

NotFoundError (see above for traceback): Key Variable_10 not found in checkpoint

原因:

保存在checkpoint中的变量如weights的变量名为weights,
预测第一张图片时,weights的name="weights";
预测第二张时,模型再次加载,weights也第二次加载,但是name="weights"已经存在,name会变成"weights_2",但是找不到这个变量名,于是报错。
详细见https://blog.csdn.net/l18930738887/article/details/69808692

解决方案:

在每一次预测结束后加上tf.reset_default_graph()

读取多张图片并进行预测

test_pics_dir ="./test_pics"
files = os.listdir(test_pics_dir)
for file in files:
    pre_pic_path = test_pics_dir+"/"+file
    image = predict(pre_pic_path)
    y_predict = get_result(image)
    print("真实值:",p_true,"预测值为:",y_predict)
    #重新定义默认图
    tf.reset_default_graph()

tf.reset_default_graph函数用于清除默认图形堆栈并重置全局默认图形

上一篇下一篇

猜你喜欢

热点阅读