caffe中预训练的imagenet模型的对象类别 [英] Object categories of pretrained imagenet model in caffe

问题描述

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

I'm using the pretrained imagenet model provided along the caffe (CNN) library ('bvlc_reference_caffenet.caffemodel'). I can output a 1000 dim vector of object scores for any images using this model.
However I don't know what the actual object categories are. Did someone find a file, where the corresponding object categories are listed?

推荐答案

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

You should look for the file 'synset_words.txt' it has 1000 line each line provides a description of a different class.

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

For more information on how to get this file (and some others you might need) you can read this.

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

If you want all the labels to be ready-for-use in Matlab, you can read the txt file into a cell array (a cell per class):

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

这篇关于caffe中预训练的imagenet模型的对象类别的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！