Как использовать более общие метки при использовании Caffe для классификации

Я использую предварительно обученный AlexNet в Caffe для классификации. Однако результат классификации составляет около 1000 классов из файла 'synset_words.txt'. Есть ли способ использовать более общие классы?

Например, вместо того, чтобы выводить тип собаки на картинке, я просто хочу, чтобы вывод был «собака».

Я знаю, что в классах есть иерархия, но есть ли способ получить эту иерархию и использовать ее для получения более общих меток?

Благодарность :)


deep-learning neural-network classification caffe pycaffe
person BF Mohtedi    schedule 30.03.2016    source источник

Ответы (1)


arrow_upward
2
arrow_downward

Цитата из документации по API image-net:

Иерархия WordNet
следующий файл обеспечивает все отношения «is-a», предоставляемые WordNet 3.0. Каждая строка представляет собой пару родитель-потомок.

Вам необходимо загрузить файл 'wordnet.ia_a.txt', проанализировать его и построить иерархию сети слов.

person Shai    schedule 31.03.2016