如何修改TensorFlow Object Detection API代码输出置信度>10%的检测框

嘿，这个调整其实超简单，只要找到代码里控制绘制检测框的阈值参数就行，具体步骤如下：

• 打开你的Object Detection Tutorial笔记本，定位到可视化检测结果的代码块，也就是调用vis_util.visualize_boxes_and_labels_on_image_array函数的部分。
• 这个函数里有一个关键参数min_score_thresh，它就是用来控制只绘制置信度高于该值的检测框的。默认设置的是0.5（对应50%的置信度）。
• 把这个参数的值改成0.1，示例代码如下：

  vis_util.visualize_boxes_and_labels_on_image_array(
      image_np,
      np.squeeze(boxes),
      np.squeeze(classes).astype(np.int32),
      np.squeeze(scores),
      category_index,
      use_normalized_coordinates=True,
      line_thickness=8,
      min_score_thresh=0.1)  # 这里把默认的0.5改成0.1
  

• 运行修改后的代码，就能看到所有置信度大于10%的目标检测框被绘制出来了。

这个参数的核心作用就是过滤掉置信度低于设定值的检测结果，调整数值后就能直接得到你想要的效果啦～

内容的提问来源于stack exchange，提问作者Ajinkya