Tensorflow对象检测API-可视化区域建议

时间:2018-07-08 13:27:08

标签: python tensorflow computer-vision object-detection tensorboard

我希望能够使用Tensorflow Object Detection API(最好在Tensorboard中)可视化Faster-RCNN(例如Resnet101_coco)提出的区域建议。有什么办法吗?

1 个答案:

答案 0 :(得分:0)

您可以在评估期间(即在运行object_detection/eval.py脚本时)在张量板上可视化检测到的对象,您需要向配置文件中添加num_visualizations键,例如

eval_config: {
  num_examples: 20000
  num_visualizations: 16
  min_score_threshold: 0.15
  # Note: The below line limits the evaluation process to 10 evaluations.
  # Remove the below line to evaluate indefinitely.
  max_evals: 1
}

运行评估后,您应该能够在Tensorboard中看到一个图像选项卡,其中显示检测到的对象。您可以调整IoU阈值(min_score_threshold)来更改显示的检测次数。