VGG:分类顺序



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

您可以在ILSVRC中获得1000个类中每个类的概率的1000维矩阵。

然而,我找不出矩阵中的哪一列对应于哪个类。

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

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

但矩阵似乎不是按这个顺序排列的。

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

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

最新更新