VGG:类别的顺序?

时间:2016-04-15 17:11:33

标签: deep-learning computer-vision caffe imagenet vgg-net

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

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

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

例如,我想知道final_prob [2]是否与人或车相对应。

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

但矩阵似乎没有按顺序排列。

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

1 个答案:

答案 0 :(得分:1)

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