1

我使用TensorFlow 对象检测 API在自定义数据集上训练模型。tensorflow 对象检测 API 还允许在测试集上评估经过训练的模型,并以COCO eval 格式给出结果。可以在此处看到评估的示例输出:

TensorFlow Object Detection API 的评估输出报告 MSCOCO 指标

在这张图中,有mAP@0.5 IoU和mAP@0.75 IoU等结果。它还报告 AR(平均召回率)。它还报告了一个整体的mAP。

训练完我的模型后,我将模型转换为 Tensorflow Lite,并在模型上应用了训练后量化。现在我想在我的测试集上为我的 tflite 模型计算类似的指标。

有人可以指导我如何去做吗?我试图了解如何使用COCOAPI来做到这一点,但到目前为止我不确定如何。

一个相关的问题是,有人可以告诉我在哪里可以找到COCOAPI的文档吗?我到处搜索,我似乎找不到它。

谢谢

4

0 回答 0