neural-network - 使用 Caffe 进行分类时如何使用更通用的标签

Question

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

例如，我不希望输出图片中的狗类型，而是希望输出为“狗”。

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

谢谢 ：）

Answer 1

引用image-net API 文档：

WordNet Hierarchy
以下文件提供了 WordNet 3.0 提供的所有“is-a”关系。每行是一个父子对。

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