python – 在Tensorflow Object Detection API中打印类名和分数

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 在Tensorflow Object Detection API中打印类名和分数，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含2663字，纯文字阅读大概需要4分钟。

内容图文

我正在使用Tensorflow对象检测API一切正常但我想打印一个dict或数组,其格式如下{Object name,Score}或类似的东西,我需要的是对象名称和分数.

我尝试使用以下代码：

with detection_graph.as_default():
  with tf.Session(graph=detection_graph) as sess:
    # Definite input and output Tensors for detection_graph
    image_tensor = detection_graph.get_tensor_by_name('image_tensor:0')
    # Each box represents a part of the image where a particular object was detected.
    detection_boxes = detection_graph.get_tensor_by_name('detection_boxes:0')
    # Each score represent how level of confidence for each of the objects.
    # Score is shown on the result image, together with the class label.
    detection_scores = detection_graph.get_tensor_by_name('detection_scores:0')
    detection_classes = detection_graph.get_tensor_by_name('detection_classes:0')
    num_detections = detection_graph.get_tensor_by_name('num_detections:0')
    for image_path in TEST_IMAGE_PATHS:
      image = Image.open(image_path)
      # the array based representation of the image will be used later in order to prepare the
      # result image with boxes and labels on it.
      image_np = load_image_into_numpy_array(image)
      # Expand dimensions since the model expects images to have shape: [1, None, None, 3]
      image_np_expanded = np.expand_dims(image_np, axis=0)
      # Actual detection.
      (boxes, scores, classes, num) = sess.run(
          [detection_boxes, detection_scores, detection_classes, num_detections],
          feed_dict={image_tensor: image_np_expanded})
      print ([category_index.get(value) for index,value in enumerate(classes[0]) if scores[0,index] > 0.5])

      threshold = 0.5 # in order to get higher percentages you need to lower this number; usually at 0.01 you get 100% predicted objects
      print(len(np.where(scores[0] > threshold)[0])/num_detections[0])

这部分是有效的

  print ([category_index.get(value) for index,value in enumerate(classes[0]) if scores[0,index] > 0.5])

它正在打印[{‘name’：’computer’,’id’：1}]他们是否能以任何方式将该对象的分数添加到字典？

我在他们使用的Stackoverflow上看到了另一个问题：

 threshold = 0.5 # in order to get higher percentages you need to lower this number; usually at 0.01 you get 100% predicted objects
 print(len(np.where(scores[0] > threshold)[0])/num_detections[0])

这给了我Tensor(“truediv：0”,dtype = float32)但是如果它工作的话虽然它还不够,因为我没有对象的名字.

谢谢

解决方法:

所以这是适合我的解决方案. (如果您仍在寻找解决方案,那就是)

# The following code replaces the 'print ([category_index...' statement
objects = []
for index, value in enumerate(classes[0]):
  object_dict = {}
  if scores[0, index] > threshold:
    object_dict[(category_index.get(value)).get('name').encode('utf8')] =                         scores[0, index]
    objects.append(object_dict)
print objects

内容总结

以上是互联网集市为您收集整理的python – 在Tensorflow Object Detection API中打印类名和分数全部内容，希望文章能够帮你解决python – 在Tensorflow Object Detection API中打印类名和分数所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端

---