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

Posted 2023-03-13

【中文标题】使用 Coffee 进行分类时如何使用更通用的标签

【英文标题】：How to use more generalized lables when using Caffe for classification

【发布时间】：2016-07-19 04:45:22

【问题描述】：

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

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

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

谢谢:)

【参考方案1】：

引用image-net API documentation:

WordNet 层次结构following file 提供了 WordNet 3.0 提供的所有“is-a”关系。每行是一个父子对。

您需要下载'wordnet.ia_a.txt' 文件，对其进行解析并构建wordnet 层次结构。