Faster RCNN 的输出

Output of Faster RCNN

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

是的,Faster RCNN class单独确定每个检测,因此对于给定的 class 可以有任意数量的检测。您可以过滤掉除您感兴趣的 class 之外的所有检测,然后寻找置信度最高的 3 个检测。

您可以在可视化工具中非常轻松地做到这一点 FiftyOne:

R-CNN model predictions in FiftyOne