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

Question

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

标签： pythontensorflowcomputer-visionobject-detectiontensorboard

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) 以改变显示的检测数量。