0

我想可视化 ResNet-101 模型中 Region Proposal Network 提出的边界框。我正在使用detectron2/configs/COCO-InstanceSegmentation/mask_rcnn_R_101_FPN_3x配置。

我想要像这样可视化的负锚框和正锚框:

在此处输入图像描述

训练后我有模型对象:( model = trainer.model这给了我一个 GeneralizedRCNN 对象)

以及从 cv2 读取的图像变量: img = cv2.imread(filename)

任何关于我如何继续这样做的指导将不胜感激。

4

0 回答 0