如何找到训练 Caffe 网络的 classes/objects 列表(以及它们的索引)?
How to find list of classes/objects that Caffe net was trained with (and also their indices)?
Caffe Model Zoo (http://caffe.berkeleyvision.org/model_zoo.html) 提供了各种我想使用的预训练 DNN 模型。但是,我正在努力寻找 类 的列表和这些网络训练的对象及其相应的索引。
有谁知道在哪里可以找到这个?我查看了自述文件和实际的 protobuf,但一无所获。
谢谢!
model zoo 中的模型通常 trained/tested 在选定的几个公开可用的基准测试中。我相信每个模型都清楚地说明了它 trained/tested 的基准。知道使用了什么基准,您可以直接从基准网站访问标签列表。
例如,在 ImageNet (aka ILSVRC challenge) are trained to classify each image into one of 1000 classes (last layer's num_output: 1000
). You can find more information about the classes of this benchmark and here.
上训练的模型
一些模型是在 CIFAR 小图像上训练的。该基准有两个 "flavors":CIFAR10 和 CIFAR100,分别将每个图像分类为 10 或 100 类(最后一层的 num_output
为 10 或 100)。您可以在基准测试网页上找到 类 的列表。
除了分类模型之外,还有检测模型,经过训练可以对每个图像中的多个对象进行分类和。这些模型通常在 MSCOCO which has 80 classes (if I'm not mistaken), and PASCAL VOC detection challenge 等基准上进行训练,其中有 20 类.
最重要的是,您需要查看每个模型是在哪个基准上训练的,然后从基准网页上获取标签。
Caffe Model Zoo (http://caffe.berkeleyvision.org/model_zoo.html) 提供了各种我想使用的预训练 DNN 模型。但是,我正在努力寻找 类 的列表和这些网络训练的对象及其相应的索引。
有谁知道在哪里可以找到这个?我查看了自述文件和实际的 protobuf,但一无所获。
谢谢!
model zoo 中的模型通常 trained/tested 在选定的几个公开可用的基准测试中。我相信每个模型都清楚地说明了它 trained/tested 的基准。知道使用了什么基准,您可以直接从基准网站访问标签列表。
例如,在 ImageNet (aka ILSVRC challenge) are trained to classify each image into one of 1000 classes (last layer's num_output: 1000
). You can find more information about the classes of this benchmark
一些模型是在 CIFAR 小图像上训练的。该基准有两个 "flavors":CIFAR10 和 CIFAR100,分别将每个图像分类为 10 或 100 类(最后一层的 num_output
为 10 或 100)。您可以在基准测试网页上找到 类 的列表。
除了分类模型之外,还有检测模型,经过训练可以对每个图像中的多个对象进行分类和。这些模型通常在 MSCOCO which has 80 classes (if I'm not mistaken), and PASCAL VOC detection challenge 等基准上进行训练,其中有 20 类.
最重要的是,您需要查看每个模型是在哪个基准上训练的,然后从基准网页上获取标签。