Faster RCNN 的输出

Question

是否可以在 Faster RCNN 中获得单个类的 3 个边界框？ 在我的工作中，我想从 Faster RCNN 获得 3 条最佳建议。 例如假设我的模型正在寻找树上的苹果。所以一棵树上可能有这么多苹果，但我想得到前 3 个作为输出。

Answer 1

是的，Faster RCNN 会单独对每个检测进行分类，因此给定类别可以有任意数量的检测。您可以过滤掉除您感兴趣的类之外的任何检测，然后寻找置信度最高的 3 个检测。

您可以在可视化工具FiftyOne 中轻松做到这一点：

R-CNN model predictions in FiftyOne