TF:检查点文件中找不到张量名称

时间:2018-02-28 12:02:38

标签: python tensorflow machine-learning

我一直在努力尝试使用此回购https://github.com/tensorflow/models/tree/master/research/slim中给出的预训练模型。我正在使用此网https://github.com/tensorflow/models/blob/master/research/slim/nets/inception_v3.py和ILSVRC-2012-CLS图像分类数据集的存储库主页面中给出的检查点文件。我正在为Inception V3下载一个。

我只是想做一个预测,所以我的代码如下:

placeholder = tf.placeholder(tf.float32, shape=[None, input_width, input_height, 3])

# Select the model
model = inception(placeholder)

# Predict
with tf.Session() as sess:
    tf.train.Saver().restore(sess, "./model/inception_v3.ckpt")
    init_op = tf.initialize_all_variables()
    predictions = sess.run(model, feed_dict={placeholder: input})

但我收到错误"NotFoundError (see above for traceback): Tensor name "InceptionV3/AuxLogits/Conv2d_2a_5x5/biases" not found in checkpoint files ./model/inception_v3.ckpt"

所以我不知道ckpt文件是否未更新为Tensor名称,或者我是否做错了。我检查了inception_v3.ckpt文件中的所有Tensor Names,但是我找不到给出错误的张量名称。

提前致谢。

0 个答案:

没有答案