问题
I'm using the pretrained imagenet model provided along the caffe (CNN) library ('bvlc_reference_caffenet.caffemodel'
). I can output a 1000 dim vector of object scores for any images using this model.
However I don't know what the actual object categories are. Did someone find a file, where the corresponding object categories are listed?
回答1:
You should look for the file 'synset_words.txt'
it has 1000 line each line provides a description of a different class.
For more information on how to get this file (and some others you might need) you can read this.
If you want all the labels to be ready-for-use in Matlab, you can read the txt file into a cell array (a cell per class):
C = textread('/path/to/synset_words.txt','%s','delimiter','\n');
来源:https://stackoverflow.com/questions/30259911/object-categories-of-pretrained-imagenet-model-in-caffe