评估 TensorFlow 对象检测模型

Question

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

Answer 1

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

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

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

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

Answer 2

您可以使用混淆矩阵在测试数据集上评估您的模型。 在数据集上训练模型后，导出推理图进行评估。 找到可以帮助您逐步进行评估的附加链接。 祝你好运！ confusion_matrix