运行训练有素的卷积神经网络

时间:2017-04-05 23:05:22

标签: machine-learning conv-neural-network

所以这是我的问题。我已经训练了一个卷积神经网络,使用tensorflow将图像分为两类。我现在想知道如何使用神经网络中的权重并在未标记的随机图像上进行测试。在tensorflow中是否有一个函数可以做到这一点,还是我现在应该自己运行卷积?

1 个答案:

答案 0 :(得分:0)

完成培训后,您可以创建

feed_dict_unlabeled = {x: x_unlabeled}

现在,使用代码中定义的y_pred_cls,如下所示:

y_pred_cls = tf.argmax(y_pred, dimension=1)

你可以做到

y_labels = session.run(y_pred_cls, feed_dict=feed_dict_unlabeled)

找到未标记数据的标签。

此外,以下是您可能会发现有用的类似方案的讨论:Python/Tensorflow - I have trained the convolutional neural network, how to test it?