2017-05-23 7 views
0

テンソルフローモデルを10個の画像でバッチサイズでテストしようとしています。出力として特定の領域のセグメンテーションを取得しようとしています。ここで私はテンソルフローモデルでの画像のテスト

def run(model_path, image_path,graph_path ,output_path): 
with tf.Session() as sess: 
    model_saver = tf.train.import_meta_graph(graph_path, clear_devices=True) 
    model_saver.restore(sess,tf.train.latest_checkpoint(LOGDIR)) 
    sess.run(tf.global_variables_initializer()) 
    graph = tf.get_default_graph() 
    x=tf.placeholder(tf.float32,shape=[10, 400, 600, 3]) 
    output = graph.get_tensor_by_name("prediction:0") 
    print("Model restored.") 
    print('Initialized') 
    temp=[] 
    import glob 
    from scipy import misc 
    for file in os.listdir(image_path): 
    if file.endswith(".jpg"): 
     img = misc.imread(image_path + file) 
     img = img.astype('Float32') 
     img = np.resize(img,(400,600,3)) 
     temp.append(img) 
    prediction = sess.run(output, feed_dict={x:temp}) 
    cv2.imwrite(output_path, prediction * 255) 

モデルをロードするために使用しています。しかし、私は次のエラーを取得するコードされています

InvalidArgumentError (see above for traceback): You must feed a value for placeholder tensor 'x' with dtype float and shape [10,400,600,3] 
    [[Node: x = Placeholder[dtype=DT_FLOAT, shape=[10,400,600,3], _device="/job:localhost/replica:0/task:0/gpu:0"]()]] 
    [[Node: prediction/_265 = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/cpu:0", send_device="/job:localhost/replica:0/task:0/gpu:0", send_device_incarnation=1, tensor_name="edge_28_prediction", tensor_type=DT_INT32, _device="/job:localhost/replica:0/task:0/cpu:0"]()]] 

は誰がバグだった知っていがあるかのモデルをテストする別の方法があれば。

+0

ここではxを再定義していると思います。代わりに名前でフィードしてください。x = placeholder()行を削除し、{'x:0':temp}のようなものをフィードdictとして使用して何が起こるか教えてください。 –

+0

@AlWldありがとうございました! –

+0

聞いて嬉しいですが、あなたが解決したようにマークすることができればそれを答えとして加えました。 –

答えて

0

ここではxを再定義していると思います。代わりに名前を入力して、x = placeholder()行を削除し、{'x:0':temp}をフィードdictとして使用します。

関連する問題