我正在尝试部分运行检测器 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
它通过准确的蒙版和边界框为我提供了完美的结果。这怎么可能?