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

作者: 石茶
发布时间: 2024-06-16 08:14:57 (1月前)
转自：

2 条回复

0#
回复此人
荧惑 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> API提供了一个名为的教程文件 <a href="https://github.com/tensorflow/models/blob/master/research/object_detection/object_detection_tutorial.ipynb" rel="nofollow noreferrer"> object_detection_tutorial.ipynb </A> ，在这个文件中，功能 <code> run_inference_for_single_image </code> 返回检测字典 <code> output_dict </code> 其中包含密钥 <code> detection_classes </code> ，这对应于您在label_map.pbtxt文件中定义的ID。也在这个文件变量中 <code> category_index </code> 是作为字典存储的标签映射。因此，只需要获取检测到的所有对象的字符串名称，只需添加： </p> <pre> <code> string_name = [category_index[i] for i in output_dict['detection_classes']] </code> </pre> <P> 在这一行之后 <code> output_dict = run_inference_for_single_image(image_np, detection_graph) </code> 在教程中。 </p> </DIV>

编辑

登录后才能参与评论