我使用的是在带有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维输出的顺序相同。