使用 Coffee 进行分类时如何使用更通用的标签

How to use more generalized lables when using Caffe for classification

我在 Caffe 中使用预训练的 AlexNet 进行分类。但是,'synset_words.txt' 文件的分类输出约为 1000 类。有没有办法使用更通用的 类?

例如,我不想输出图片中狗的类型,而是希望输出为 "dog"。

我知道 类 中有一个层次结构,但有没有办法检索该层次结构并使用它来获取更多通用标签?

谢谢:)

引用自image-net API documentation

WordNet Hierarchy
The following file provides all "is-a" relations provided by WordNet 3.0. Each line is a parent-child pair.

您需要下载 'wordnet.ia_a.txt' 文件,解析它并构建 wordnet 层次结构。