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文件。