评估TensorFlow对象检测模型

Question

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

Answer 1

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

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

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

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

Answer 2

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