ios - 为什么物体检测会导致找到多个物体？

Question

我用 CreateML 训练了一个对象检测器，当我在 CreateML 中测试模型时，我得到了大量已识别的对象：

笔记：

• 该模型在一个包含约 30 张图像的小型数据集上进行了训练，该特定标签face-gendermale出现了约 20 次。
• 每个训练图像有 1-3 个标记对象。
• 共有 5 个标签。

问题：

• 这是预期的还是模型有问题？
• 如果这是预期的，我应该如何评估这些多个结果，甚至计算模型中找到的对象数量？

在Apple 开发者论坛中交叉发布。男人的照片© Jason Stitt | Dreamstime.com

Answer 1

一个典型的对象检测模型对每张图像进行大约 1000 次预测（尽管它可能更多取决于模型架构）。大多数这些预测的置信度都非常低，因此被过滤掉了。然后剩下的那些通过非最大抑制（NMS）发送，它去除了重叠太多的边界框。

在您的情况下，NMS 的阈值似乎太低（或太高），因为许多重叠框仍然存在。

但是，该模型似乎还没有得到很好的训练，可能是因为您使用的图像很少。