提取检测到的类名 - tensorflow对象检测API

Question

我的问题是对this提供非常丰富信息的帖子的跟进。我要更进一步，只打印检测到的类名。

正如在帖子中所回答的，我目前能够使用代码行[{'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))