首页 > 解决方案 > 更改 TensorFlow 2 Object Detection API 的性能指标

问题描述

我在哪里可以找到配置文件?

默认指标基于 Pascal VOC 评估中使用的指标。

要使用 COCO 对象检测指标,请将 metrics_set: "coco_detection_metrics" 添加到配置文件中的 eval_config 消息中。

要使用 COCO 实例分段指标,请将 metrics_set: "coco_mask_metrics" 添加到配置文件中的 eval_config 消息中。

https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/install.html

标签: tensorflow

解决方案


如果您使用模型动物园中的预训练模型,则配置文件是pipeline.config每个模型存档中的配置文件。如果您正在构建自己的网络,则必须pipeline.config按照指南编写。因此,只需在COCO 部分编辑您的pipeline.config添加,如果您想要 PASCAL-VOC,则相同。metrics_set: "coco_detection_metrics"eval_config


推荐阅读