Я использую модель vgg на основе keras в качестве классификатора изображений. С этой веб-страницы:
https://machinelearningmastery.com/use-pre-trained-vgg-model-classify-objects-photographs/
Но мне интересно узнать родительскую категорию предсказания. Например, если модель предсказывает собаку, я хотел бы знать, что ее родительская категория - животное. Я не знаю, есть ли способ использовать дерево Imagenet для этой проблемы?
ImageNet использует WordNet для создания иерархии классов. Вы можете получить доступ к набору объектов следующим образом:
import nltk
from nltk.corpus import wordnet
nltk.download('wordnet')
dog = wordnet.synsets('dog')[0]
Чтобы получить доступ к родителям dog, позвоните:
dog.hypernyms()
Он может вернуть более одного результата, что проблематично.
Что вы можете сделать сейчас, так это проверить, является ли какой-либо из этих родителей реальным ImageNet классом. Чтобы получить wnid набора символов, просто вызовите dog.offset().
Чтобы повернуть процесс вспять:
from nltk.corpus import wordnet as wn
wn.synset_from_pos_and_offset('n',2084071)
