Tensorflowのリストアチェックポイント - テンソル名が見つかりません

Question

Googleが提供するアーキテクチャとチェックポイントでInceptionv3 Tensorflowモデルを実行しようとしていますhere。

私の問題は私のスクリプトは次のエラーでsaver.restore(sess, "./inception_v3.ckpt")にクラッシュするということである。ここでは

tensorflow.python.framework.errors.NotFoundError: Tensor name "InceptionV3/Mixed_5b/Branch_1/Conv2d_0b_5x5/biases" not found in checkpoint files ./inception_v3.ckpt 

は私のコードです：

import tensorflow as tf 
import inception_v3 

with tf.Session() as sess: 
    image = tf.read_file('./file.jpg') 
    # code to decode, crop, convert jpeg 
    eval_inputs = tf.pack([image]) 
    logits, _ = inception_v3.inception_v3(eval_inputs, num_classes=1001, is_training=False) 
    sess.run(tf.initialize_all_variables()) 

    saver = tf.train.Saver() 
    saver.restore(sess, "./inception_v3.ckpt") 

私は他のチェックポイント/モデルの組み合わせと同じエラーが出ますこれは私のコードの問題でなければなりません。私が間違って何をしているかわからない。

ありがとうございました

Answer 1

確かに、チェックポイントファイルにはこのテンソルが含まれていません。 githubにバグを報告できますか？

Answer 2

あなたは、このようなinception_v3_arg_scope()によって返さarg_scope()内inception_v3()を呼び出す必要があります：

import tensorflow as tf 
import tensorflow.contrib.slim as slim 
from nets.inception_v3 import inception_v3, inception_v3_arg_scope 

height = 299 
width = 299 
channels = 3 

# Create graph 
X = tf.placeholder(tf.float32, shape=[None, height, width, channels]) 
with slim.arg_scope(inception_v3_arg_scope()): 
    logits, end_points = inception_v3(X, num_classes=1001, 
             is_training=False) 
predictions = end_points["Predictions"] 
saver = tf.train.Saver() 

X_test = ... # your images, shape [batch_size, 299, 299, 3] 

# Execute graph 
with tf.Session() as sess: 
    saver.restore(sess, "./inception_v3.ckpt") 
    predictions_val = predictions.eval(feed_dict={X: X_test}) 

predicted_classes = np.argmax(predictions_val, axis=1) 

私ははっきり建設段階と実行フェーズの分離をお勧めします。ちょうどWeb上のランダムな写真をテストしたところ、正常に動作しました。 :)