首页 文章

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

提问于
浏览
2

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

例如,不是在图片中输出狗的类型,我只想让输出为“狗” .

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

谢谢 :)

1 回答

  • 2

    引自image-net API documentation

    WordNet层次结构以下文件提供WordNet 3.0提供的所有“is-a”关系 . 每一行都是父子对 .

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

相关问题