如何在TensorFlow Object Detection API中提取并仅打印检测类名

嘿，这个小调整超简单的！你现在用print(category_index.get(value))得到的是一个包含字典的列表，只要精准定位到字典里的name字段就行啦。

情况1：每次仅检测到一个目标 #

直接通过索引取出列表里的第一个字典，再获取name对应的值：

print(category_index.get(value)[0]['name'])

这样终端就会直接输出bottle这类纯类名啦。

情况2：可能检测到多个目标 #

如果存在多个检测结果，咱们可以循环遍历列表，逐个输出类名：

detections = category_index.get(value)
for det in detections:
    print(det['name'])

这样每个检测到的类名都会单独一行打印出来。

内容的提问来源于stack exchange，提问作者Veejay