我正在使用Caffe(CNN)库提供的预训练ImageNet模型('bvlc_reference_caffenet.caffemodel'
)。使用这个模型,我可以为任何图像输出一个1000维的对象得分向量。
然而,我不知道实际的对象类别是什么。有人找到一个列出相应对象类别的文件了吗?
回答:
你应该查找文件'synset_words.txt'
,它有1000行,每行提供一个不同类别的描述。
关于如何获取这个文件(以及你可能需要的其他一些文件)的更多信息,你可以阅读这里。
如果你希望在Matlab中准备好使用所有标签,你可以将txt文件读取到一个单元数组中(每个类别一个单元):
C = textread('/path/to/synset_words.txt','%s','delimiter','\n');