具有奇怪检测结果的Tensorflow对象检测API

Question

我已经使用rfcn_resnet101模型在我自己的数据集中训练了带有1个类的tensorflow对象检测api。 首先，我使用浣熊数据集并对其进行了264600次训练，其检测结果很奇怪，可以检测到物体，但是右侧的盒子周围还有其他一些小盒子。

然后我使用另一个包含一个类的数据集，并且数据集中有8万张图像，我遇到了熟悉的现象。 我很困扰。

有没有人遇到过同样的情况？ 我该怎么做才能解决这个问题？ 提前致谢！

Answer 1

我在PASCAL VOC数据集上具有相同的行为。 我还没有解决它，因为我只是为某种概念验证系统实现了一个模型。 我的猜测是，如果IoU大于或等于定义的阈值，则模型会预测提案区域并接受它们。 因此，设置nms_iou_threshold可能会解决该问题。

这种采用似乎也适合您的示例。 所有预测的边界框似乎与groundtruth框都有一个IoU。