caffe中预训练imagenet模型的对象类别

时间:2015-05-15 12:46:17

标签: machine-learning neural-network deep-learning caffe matcaffe

我使用沿caffe(CNN)库('bvlc_reference_caffenet.caffemodel')提供的预训练imagenet模型。我可以使用此模型为任何图像输出1000暗淡的对象分数向量 但是,我不知道实际的对象类别是什么。有人找到了一个文件,列出了相应的对象类别吗?

1 个答案:

答案 0 :(得分:5)

你应该查找文件'synset_words.txt'它有1000行,每行提供不同类的描述。

有关如何获取此文件(以及您可能需要的其他文件)的更多信息,请阅读this

如果您希望所有标签都可以在Matlab中使用,您可以将txt文件读入单元格数组(每个类的单元格):

C = textread('/path/to/synset_words.txt','%s','delimiter','\n');