在图像分割中获得单个测试图像的准确性

时间:2018-05-17 07:04:13

标签: tensorflow machine-learning image-segmentation convolutional-neural-network

我在Tensorflow中使用CNN进行图像分割。我知道如何计算训练精度

 #compute the accuracy
    correct_prediction  = tf.equal(tf.argmax(flat_logits, 1), tf.argmax(y,1))
    accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))



[train_accuracy] = sess.run([accuracy], feed_dict={x: batch_x, y:batch_y})

是否可以计算每个测试图像的准确度的准确性?

1 个答案:

答案 0 :(得分:1)

是的,有可能。你只需写下来就可以做到:

test_accuracy = sess.run(accuracy, feed_dict={x: x_test, y:y_test})

其中x_test是您的单个测试图像(比如大小[1,宽度,高度,深度],y_test是相应的输出。