imagenet_hypernyms.py

gistfile1.txt

import sys
import nltk
from nltk.corpus import wordnet as wn

nltk.download("wordnet")


def find_hypernyms(synset_name):
    synset = wn.synset(synset_name)
    hypernyms = synset.hypernyms()
    if not hypernyms:
        return [synset_name]
    return [synset_name] + find_hypernyms(hypernyms[0].name())


def hyper2text(word):
    word = word.split(".")[0]
    return word.lower().replace("_", " ")


def guess(word):
    word = word.lower().replace(" ", "_")
    hypernym_chain = find_hypernyms(f"{word}.n.01")
    print(len(hypernym_chain))
    print(hypernym_chain)
    third = hypernym_chain[int(len(hypernym_chain) / 4)]
    caption = f"The image seems to display a {hyper2text(third)}. Maybe a {hyper2text(hypernym_chain[0])}?"
    return caption


print(guess(sys.argv[-1]))