更快的RCNN

问题描述

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

解决方法

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

您可以在可视化工具FiftyOne中非常轻松地完成此操作：

R-CNN model predictions in FiftyOne