我一直在努力尝试使用此回购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,但是我找不到给出错误的张量名称。
提前致谢。