tensorflow - tensorflow对象检测评估错误

Question

当我使用模型检查测试数据集上的 mAP 时，出现以下错误：

INFO:tensorflow:Restoring parameters from /home/aurora/workspaces/PycharmProjects/tensorflow/tensorflow_object_detection/outputs/model.ckpt-278075
INFO:tensorflow:Restoring parameters from /home/aurora/workspaces/PycharmProjects/tensorflow/tensorflow_object_detection/outputs/model.ckpt-278075
WARNING:root:The following classes have no ground truth examples: 0
/home/aurora/workspaces/PycharmProjects/tensorflow/tensorflow_object_detection/object_detection/utils/metrics.py:145:                RuntimeWarning: invalid value encountered in true_divide
num_images_correctly_detected_per_class / num_gt_imgs_per_class)

我检查了 test.tfrecords，每个图像都有真实边界框。我该如何解决这个问题？谢谢。

Answer 1

我遇到了类似的错误，并且在该错误上被困了很多天。我可以通过编辑 label.pbtxt 文件来解决该错误。你能显示你的标签（.pbtxt）文件吗？我的标签文件是：（包含 3 个标签）

item {
  id: 1
  name: 'tree'

  id: 2
  name: 'water body'

  id: 3
  name: 'building'
}

然后我将其更改为：

item {
  id: 1
  name: 'tree'
     }

item {
  id: 2
  name: 'water body'
     }

item {
  id: 3
  name: 'building'
     }

这在我的情况下有效。查看您在模型配置文件中引用的 .pbtxt 文件。