Tensorflow Object Detection API - 可视化区域建议

Question

我希望能够使用 Tensorflow 对象检测 API（最好是在 Tensorboard 中）可视化由 Faster-RCNN（如 Resnet101_coco）提出的区域建议。有什么办法吗？

Answer 1

您可以在评估期间可视化 tensorboard 中检测到的对象（即在运行 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) 以改变显示的检测数量。