API提供了一个名为的教程文件 object_detection_tutorial.ipynb ,在这个文件中,功能 run_inference_for_single_image 返回检测字典 output_dict 其中包含密钥 detection_classes ,这对应于您在label_map.pbtxt文件中定义的ID。也在这个文件变量中 category_index 是作为字典存储的标签映射。因此,只需要获取检测到的所有对象的字符串名称,只需添加:
run_inference_for_single_image
output_dict
detection_classes
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) 在教程中。
output_dict = run_inference_for_single_image(image_np, detection_graph)