tensorflow - 评估 TensorFlow 对象检测模型

Question

如何以简单易懂的方式评估我的对象检测模型，我使用了 TensorFlow 的对象检测 API，但我不了解 Tensorboard 图。我可以手动评估它吗？有什么帮助吗？:(

标签： tensorflowobject-detectiontensorboardevaluate

Answer 1

欢迎来到 StackOverflow！简而言之，是的，你可以。然而，实现您的目标可能非常耗时。以下是您可能想要遵循的步骤（假设您对 Tensorflow 图和会话有一些基本了解，否则请更新您的问题）：

1. 通过HERE将模型导出到冻结图（*.pb 文件）。此步骤将为您提供一个开箱即用的模型，您可以在不依赖任何对象检测 API 的情况下加载该模型。

2. 编写一个脚本来加载您的模型（冻结图）并执行评估。一些说明可以从这里找到。确保使用Netron等工具检查冻结图的输入和输出节点名称。

3. 一旦可以执行评估，您就可以在自己的数据集上编写指标，例如 mAP，并循环遍历所有图像以执行所需的评估。