我对目标检测真的很陌生,如果这个问题看起来太明显的话,我很抱歉。我在FasterRCNN上有一个经过训练的detectron2模型,用于检测对象,并且我正在尝试提取每个检测对象的特征,以用于模型预测的输出。可用的教程似乎是为了寻找ROI,并做出新的预测,以及它们的框和特性。我有来自推理的框,我只需要提取每个框的特征。我添加了下面的代码。= [features[f] for f in predictor.model.roi_heads.in_features]
proposal_rois = predict