如何自定义掩码 r-cnn 的 display_instances() 函数的输出

问题描述

如何使用 display_instances() 函数只绘制没有边界框的蒙版，反之亦然？

visualize.display_instances(image=image,Boxes=r['rois'],masks=r['masks'],class_ids=r['class_ids'],class_names=class_names)

解决方法

将visualize.py 更新到以下版本，其中可以选择显示掩码或bbox。 updated visualize.py here

但是标签部分需要修改如下：

# Label
        if not captions:
            class_id = class_ids[i]
            score = scores[i] if scores is not None else None
            label = class_names[class_id]
            caption = "{} {:.3f}".format(label,score) if score else label
            ax.text(x1,y1 + 8,caption,color="none",size=11,backgroundcolor="none") # to  remove captions change to: color= "none"

        else:
            caption = captions[i]
            ax.text(x1,color='w',backgroundcolor="none") # to  remove captions change to: color= "none"