Извлечение обнаруженного имени класса - API обнаружения объектов tenorflow

Question

Мой вопрос - продолжение до этой чрезвычайно информативной ветки.Я должен был сделать этот шаг дальше и распечатать только обнаруженное имя класса.

Как ответили в теме, в настоящее время я могу получить: [{'id: 4 , 'name': 'bottle'}] используя строку кода print[category_index.get(value)

Я пытался выделить только обнаруженное имя класса, чтобы я мог напечататьвыход, который просто bottle на терминале.Какие изменения мне нужно внести в мою линию печати, чтобы выполнить это

Answer 1

Попробуйте это:

def GetClassName(data):
   for cl in data:
    return cl['name']

#data processed

data = [category_index.get(value) for index,value in enumerate(classes[0]) if scores[0,index] > 0.9]

print(GetClassName(data))