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

Question

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

Answer 1

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

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

<小时/>

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

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