VGG:类别顺序?

VGG: order of categories?

我正在使用在 ILSVRC 上训练的 VGG-16 网络和 Fast-RCNN。

在 ILSVRC 中,您可以获得 1,000 个 class 中每个概率的 1000 维矩阵。

但是,我无法找出矩阵中的哪一列对应于哪个class。

比如我想知道final_prob[2]对应的是人还是车

ILSVRC 每年有 classes 列表 (http://image-net.org/challenges/LSVRC/2014/browse-synsets)

但矩阵似乎不是这个顺序。

在哪里可以找到矩阵中的条目是如何映射到 classes 的?

如果您使用 Caffe 中的预训练模型之一,运行 data/ilsvrc12/get_ilsvrc_aux.sh 然后 类 的名称列表将在 data/ilsvrc12/synset_words.txt 中.这和模型1000维输出的顺序是一样的。