在Tensorboard中显示每个类别的指标

Question

我正在使用Tensorflow对象检测API训练对象检测模型。评估配置是这样的：

eval_config: {
  num_examples: 8000
  max_evals: 10
  num_visualizations: 20
  include_metrics_per_category: true
}

但是，张量板不显示任何每个类别指标。 我还有什么需要做的吗？

Answer 1

我认为github的pycocotools不支持此功能。我们已经在Google内部实现了此功能，但在外部无法使用。

Answer 2

如@zhichau Lu所述，由于某种原因，该功能在google之外不存在。我可以通过更改this Github答案提出的代码来使其工作。

这是我在Tensorboard中的结果的图片：

Answer 3

以下步骤可帮助我成功显示每个类别的指标（在 faster_rccn 和 mobilenet 上进行了测试）：

1。从我的git repo安装更新的pycocotools：

  

点安装   “ git + https://github.com/philippschw/cocoapi.git#egg=pycocotools&subdirectory=PythonAPI”

2。在tensorflow api中编辑此文件：

  

“ tensorflow /模型/研究/object_detetcion/metrics/coco_tools.py”：来自   第240行到244行

添加以下代码行：

# add for metrics per catergory from here
if include_metrics_per_category is True:
    self.summarize_per_category()
# add for metrics per category end here

3。编辑Tensorflow API的配置文件以包括：

  

{       metrics_set：“ coco_detection_metrics”       include_metrics_per_category：true}

例如：

eval_config: {
  num_examples: 8000
  # Note: The below line limits the evaluation process to 10 evaluations.
  # Remove the below line to evaluate indefinitely.
  max_evals: 10
  num_visualizations: 20
  metrics_set: "coco_detection_metrics" 
  include_metrics_per_category: true   
}

请注意，此问题已在tensorflow上解决，我的解决方案来自there。