所以我的问题是这样的。我已经使用TensorFlow训练了一个卷积神经网络,用于将图像分类为两个类别。现在我想知道如何使用这个神经网络的权重来测试一张未标记的随机图像。TensorFlow中有这样的功能吗,还是我需要自己运行卷积操作?
回答:
训练完成后,你可以创建
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 – 我已经训练了卷积神经网络,如何测试它?