如何查看测试集的 Caffe 预测标签？

Question

已经在github 中提出过问题，但是由于没有发布具体的答案，并且对 Caffe 框架和我要解释的其他内容进行了一些更改，我认为最好再问一次。

当 Caffe 框架学习/测试阶段发生时，在每次迭代中，它都会给我们一个 Loss 和 Accuracy 值，但 Accuracy 不是一个好的测量参数，所以我需要查看测试图像上的每个预测，一个一个 来计算其他测量参数，例如精度和召回率。

如果我跑：

import caffe

net = caffe.Net('/path/to/model_def.prototxt', '/path/to/model/weights')
out = net.forward()

它将为我们提供out['prob'] 中的第一批图像的预测，我不想预测这批图像，如何逐个获取主题？

Answer 1

结果将存储在out['prob'].data 数组中。数组的第一维对应批次。因此，要查看ith 图像的结果，请检查out['prob'].data[i] 中的值。

假设这是分类，其中每个图像都有一个为每个类别分配的值，对应于jth 类别的ith 图像的输出可以在out['prob'].data[i][j] 找到。