Tensorflow Object Detection API - 可视化区域提案

Question

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

Answer 1

您可以在评估期间(即在运行脚本时(在张量板中可视化检测到的对象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( 以改变显示的检测数量。