获得CIFAR示例（或类似）的输出概率

Question

嘿我正在试图弄清楚如何获得CIFAR示例中每个类的概率。目前我可以在输入张量中输入预测的类并返回真或假，但它不会告诉我它认为的类（满分为10）以及它对该事实的概率。有关如何在sess.run（）之后获取此信息的任何想法？

Answer 1

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