VGG：类别的顺序？

Question

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

您可以获得一个 1000 维矩阵，用于表示 ILSVRC 中 1000 个类别中每个类别的概率。

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

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

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

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

我在哪里可以找到矩阵中的条目是如何映射到类的？

Answer 1

如果您在 Caffe 中使用其中一种预训练模型，请运行 data/ilsvrc12/get_ilsvrc_aux.sh，然后类名称列表将显示在 data/ilsvrc12/synset_words.txt 中。这与模型的 1000 维输出的顺序相同。