【问题标题】:How to find list of classes/objects that Caffe net was trained with (and also their indices)?如何找到训练 Caffe 网络的类/对象列表(以及它们的索引)?
【发布时间】:2018-08-31 08:25:55
【问题描述】:

Caffe Model Zoo (http://caffe.berkeleyvision.org/model_zoo.html) 提供了各种我想使用的预训练 DNN 模型。但是,我很难找到这些网络训练过的类和对象的列表及其相应的索引。

有人知道在哪里可以找到这个吗?我查看了自述文件和实际的 protobuf,但一无所获。

谢谢!

【问题讨论】:

    标签: neural-network statistics deep-learning caffe


    【解决方案1】:

    模型库中的模型通常在选定的几个公开可用的基准上进行训练/测试。我相信每个模型都清楚地说明了它是在什么基准上进行训练/测试的。知道使用了什么基准,您可以直接从基准网站访问标签列表。

    例如,在 ImageNet(又名 ILSVRC 挑战)上训练的模型被训练为将每个图像分类到 1000 个类别中的一个(最后一层的 num_output: 1000)。您可以找到有关此基准 herehere 的类的更多信息。

    一些模型是在 CIFAR 小图像上训练的。该基准有两种“风格”:CIFAR10 和 CIFAR100,它们分别将每个图像分类为 10 或 100 个类别之一(最后一层的num_output 10 或 100)。您可以在基准网页上找到类列表。

    除了分类模型之外,还有检测模型,经过训练可以对每个图像中的多个对象进行分类定位。这些模型通常在基准上进行训练,例如 MSCOCO 有 80 个类(如果我没记错的话),PASCAL VOC detection challenge 有 20 个类。

    归根结底,您需要查看每个模型的训练基准,然后从基准网页中获取标签。

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2016-03-31
      • 2017-07-09
      • 2017-09-07
      • 2016-09-28
      • 2016-02-19
      • 2015-09-07
      • 1970-01-01
      • 2015-07-27
      相关资源
      最近更新 更多