时间戳

首页 > 解决方案 > 如何使用对象检测 API 评估自己的自定义模型？我在配置文件中写入了什么？

tensorflow - 如何使用对象检测 API 评估自己的自定义模型？我在配置文件中写入了什么？

问题描述

我有一个可以调用的自定义对象检测模型，model = MyModel()并且model.loadweights(checkpoint)我想使用对象检测 API 对其进行评估。

据我了解有两种可能性，要么我使用旧版 eval.py，我不知道在 pipeline_config 文件中放入什么

或者我使用在 model_main_tf2.py 中实现的较新版本，但是我必须将我的模型保存为 model.config 并且我也不知道放置管道文件的内容。

由于我的模型是 YOLO 模型，因此它还没有包含在样本中。

https://github.com/tensorflow/models/tree/master/research/object_detection/configs/tf2

非常感谢您的帮助！

标签： tensorflowobject-detectionobject-detection-api

解决方案

您无法使用 Object Detection API 计算 mAP，因为没有pipeline.configYolo 文件。

但是，您可以查看此repo。这是一个基于 Tensorflow 的 YoloV3 实现。他们有计算mAP的工作代码。您可以相应地修改它以计算模型的 mAP。

推荐阅读