小编DST*_*ACK的帖子

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

我正在使用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) …
Run Code Online (Sandbox Code Playgroud)

python object-detection deep-learning tensorflow

3
推荐指数
1
解决办法
4673
查看次数