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

Question

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

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

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

спасибо :)

Answer 1

Цитируя image-net API documentation:

WordNet Иерархия
following file предоставляет все "это-а" отношения, предусмотренные WordNet 3.0. Каждая строка представляет собой родительско-дочернюю пару.

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