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

Question

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

Answer 1

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

有关如何获取此文件（以及您可能需要的其他文件）的更多信息，您可以阅读此.

---

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

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