获取CIFAR示例(或类似示例)的输出概率



嘿,我正试图弄清楚如何获得CIFAR示例中每个类的概率。目前,我可以在输入张量中输入预测的类,并得到一个true或false,但它不会告诉我它认为它是什么类(满分10),以及它对这个事实的概率。有什么想法可以在sess.run()之后获得这些信息吗?

用于评估的CIFAR-10示例代码(在cifar10_eval.py中)生成(作为中间输出)logits张量,该张量包含给定图像在每个类中的对数概率。将这个张量传递给sess.run()会给你必要的信息,告诉你图像的预测类是什么。如果你计算tf.nn.softmax(logits)并将其传递给sess.run(),你会得到每个类的概率(从0到1)。

最新更新