在Caffe中预训练的Imagenet模型的对象类别

5

我正在使用caffe(CNN)库提供的预训练imagenet模型('bvlc_reference_caffenet.caffemodel')。使用该模型,我可以为任何图像输出一个1000维的对象评分向量。
然而,我不知道实际的对象类别是什么。是否有人找到了一个文件,其中列出了相应的对象类别?

1个回答

5
你应该寻找文件'synset_words.txt',它有1000行,每行提供一个不同类别的描述。
如需获得此文件(和其他可能需要的文件)的更多信息,请阅读this
如果您想让所有标签在Matlab中可以立即使用,您可以将txt文件读取到一个单元数组中(每个类别一个单元)。
C = textread('/path/to/synset_words.txt','%s','delimiter','\n');

这就是全部,谢谢 :). 顺便问一下,你有没有找到自动读取这个文本文件的最佳方法?在Matlab中可以使用"fopen(file)"然后使用"textscan"。但在这种情况下,列数必须固定(而“sysnem_words”不是这种情况)。 - mcExchange

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接