在已有tensorflow的ckpt模型时候,如何查看节点信息,网上有很多代码:
from tensorflow.python import pywrap_tensorflow checkpoint_path = '路径' reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) var_to_shape_map = reader.get_variable_to_shape_map() for key in var_to_shape_map: print("tensor_name: ", key)
上面没什么问题但是总是报错:
tensorflow.python.framework.errors_impl.NotFoundError: Unsuccessful TensorSliceReader constructor: Failed to find any matching files for ./cnn_emotion_classifier.ckpt/cnn_emotion_classifier.ckpt.data-0000-of-00001
查了一圈网上都没说清楚,其实是路径问题,这个路径不是文件路径也不是文件夹路径,先看的模型路径情况
我试了文件夹里面四个文件绝对路径填进去全部报错,原来应该是文件夹路径+模型名字+.ckpt这样表示,例如上面应该写成
D:/test/model/cnn_emotion_classifier.ckpt
但是这个cnn_emotion_classifier.ckpt并不是某个文件夹和文件名字,虽然这个是低级错误,但是对初学者却不容易解决。