deep-learning - VGG：类别的顺序？

Question

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

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

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

例如，我想知道 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 维输出的顺序相同。