【问题标题】:How to find list of classes/objects that Caffe net was trained with (and also their indices)?如何找到训练 Caffe 网络的类/对象列表(以及它们的索引)?
【发布时间】:2018-08-31 08:25:55
【问题描述】:
【问题讨论】:
标签:
neural-network
statistics
deep-learning
caffe
【解决方案1】:
模型库中的模型通常在选定的几个公开可用的基准上进行训练/测试。我相信每个模型都清楚地说明了它是在什么基准上进行训练/测试的。知道使用了什么基准,您可以直接从基准网站访问标签列表。
例如,在 ImageNet(又名 ILSVRC 挑战)上训练的模型被训练为将每个图像分类到 1000 个类别中的一个(最后一层的 num_output: 1000)。您可以找到有关此基准 here 和 here 的类的更多信息。
一些模型是在 CIFAR 小图像上训练的。该基准有两种“风格”:CIFAR10 和 CIFAR100,它们分别将每个图像分类为 10 或 100 个类别之一(最后一层的num_output 10 或 100)。您可以在基准网页上找到类列表。
除了分类模型之外,还有检测模型,经过训练可以对每个图像中的多个对象进行分类和定位。这些模型通常在基准上进行训练,例如 MSCOCO 有 80 个类(如果我没记错的话),PASCAL VOC detection challenge 有 20 个类。
归根结底,您需要查看每个模型的训练基准,然后从基准网页中获取标签。