我在Caffe中使用预先训练好的AlexNet进行分类 . 但是,分类的输出大约是 'synset_words.txt' 文件中的1000个类 . 有没有办法使用更通用的类?
'synset_words.txt'
例如,不是在图片中输出狗的类型,我只想让输出为“狗” .
我知道类中有一个层次结构,但有没有办法检索该层次结构并使用它来获得更多通用标签?
谢谢 :)
引自image-net API documentation:
WordNet层次结构以下文件提供WordNet 3.0提供的所有“is-a”关系 . 每一行都是父子对 .
您需要下载 'wordnet.ia_a.txt' 文件,解析它并构建wordnet层次结构 .
'wordnet.ia_a.txt'
1 回答
引自image-net API documentation:
您需要下载
'wordnet.ia_a.txt'
文件,解析它并构建wordnet层次结构 .