问题描述
我正在使用 Google Colab、Tensorflow 版本 1、SSD 移动网络 v2 CoCo 模型来训练自定义数据集。但是在图像上方的自定义数据集上评估 AP 和 AR = -1。不明白为什么会出现这个结果?
你能回答这个问题吗? 谢谢。
解决方法
对于任何使用 tensorflow 的对象检测模型,Average Precision 和 Average Recall 值可以是 0
但不是负值。
如果是-1
,那么原因可能如下:
-
可能是欠拟合的情况,你应该得到更多的训练数据。您应该增加模型中参数的大小或数量。尝试在现有层中添加新的神经元或训练模型的时间长一点,直到成本函数最小化。
-
您的模型可能无法区分图像,因此将相似的图像添加到您的数据集中。
-
检查输入的格式,模型可能无法以正确的格式获取输入。