我正在使用在ILSVRC上使用Fast-RCNN训练的VGG-16网络。
您可以在ILSVRC中获得1000个类别的概率的1000维矩阵。
但是,我无法找出矩阵中的哪一列对应哪个类。
例如,我想知道final_prob [2]是否与人或车相对应。
每年有ILSVRC的课程列表(http://image-net.org/challenges/LSVRC/2014/browse-synsets)
但矩阵似乎没有按顺序排列。
我在哪里可以找到矩阵中的条目如何映射到类?
答案 0 :(得分:1)
如果您正在使用Caffe中预先训练的模型之一,请运行data/ilsvrc12/get_ilsvrc_aux.sh
,然后类名称列表将在data/ilsvrc12/synset_words.txt
中。这与模型的1000维输出的顺序相同。