如何使用更快的RCNN模型获取在tensorflow中检测到的对象的名称

Question

我正在使用tensorflow的对象检测API来检测对象。我训练了模型并使其工作正常，但我想获取被检测为String的对象的名称。有人可以帮我吗？

Answer 1

API提供了一个名为object_detection_tutorial.ipynb的教程文件，在该文件中，函数run_inference_for_single_image返回一个包含关键字output_dict的检测字典detection_classes，它对应于id您在label_map.pbtxt文件中定义。同样在此文件变量category_index中，还有作为字典存储的标签图。因此，只要获取所有检测到的对象的字符串名称，只需添加：

string_name = [category_index[i] for i in output_dict['detection_classes']]

在教程中此行output_dict = run_inference_for_single_image(image_np, detection_graph)之后

。