如何在Tensorflow Object Detection API中获取预测对象的百分比值

Question

我使用此code来显示带有预测值百分比的对象检测器，但是变量num_detections是TensorVariable，例如Tensor("num_detections:0", dtype=float32)。那么如何打印预测值的百分比？

在您发表评论之前，我知道有一个类似的问题，但是答案似乎没有用。它会打印Tensor("truediv:0", dtype=float32)，我要百分比。

Answer 1

您只需要通过调用num_detections在创建的session中评估该张量sess.run。您链接的代码实际上为您做到了。

# Perform the actual detection by running the model with the image as input
(boxes, scores, classes, num) = sess.run(
    [detection_boxes, detection_scores, detection_classes, num_detections],
    feed_dict={image_tensor: image_expanded})

因此您可以简单地打印出num的值。