0

我正在尝试部分运行检测器 2 模型以可视化测试图像上的建议框。

image = ImageList.from_tensors([image])
features = model.backbone(image.tensor)
proposals, _ = model.proposal_generator(image, features)

这给了我 1000 个提案框实例:

proposals[0][i].get_fields()['proposal_boxes']

但是无论输入图像如何,这 1000 个建议框都保持不变。为什么会这样?

DefaultPredictor这很奇怪,因为如果我通过对象运行整个模型

predictor = DefaultPredictor(cfg)
outputs = predictor(im)
outputs["instances"].pred_boxes

它通过准确的蒙版和边界框为我提供了完美的结果。这怎么可能?

4

0 回答 0