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

mcE*_*nge 5 machine-learning neural-network deep-learning caffe matcaffe

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

Sha*_*hai 5

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

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


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

C = textread('/path/to/synset_words.txt','%s','delimiter','\n');
Run Code Online (Sandbox Code Playgroud)