每个边界框的R-CNN分类结果更快

Question

我用模型动物园的Faster R-CNN ResNet 101训练了一个物体检测器。

通过使用训练有素的冻结推理模型，我希望为每个预测的边界框接收分类预测。

说我有10个最终的边界框，且分数超过阈值，并且我有5个类别。我想每个边界框获得5个分类结果。在这种情况下，我将有50个值。

我当前的型号可以吗？

现在，每个框只能得到一个值。我可以在一个对象周围有多个非常相似的边界框，但这不是我想要的。