我想可视化 ResNet-101 模型中 Region Proposal Network 提出的边界框。我正在使用detectron2/configs/COCO-InstanceSegmentation/mask_rcnn_R_101_FPN_3x配置。
我想要像这样可视化的负锚框和正锚框:
训练后我有模型对象:(
model = trainer.model
这给了我一个 GeneralizedRCNN 对象)
以及从 cv2 读取的图像变量:
img = cv2.imread(filename)
任何关于我如何继续这样做的指导将不胜感激。