自学内容网 自学内容网

使用这个方法可以屏蔽掉detectron2默认的标签框

class CustomVisualizer(Visualizer):
     def overlay_instances(self, boxes=None, labels=None, masks=None, keypoints=None, assigned_colors=None, alpha=0.5):
        # 覆盖这个方法,不显示标签(labels参数不传递)
        return super().overlay_instances(boxes=boxes, labels=None, masks=masks, keypoints=keypoints, assigned_colors=assigned_colors, alpha=alpha)

# 可视化多边形和实例分割
v = CustomVisualizer(im[:, :, ::-1], metadata=self.metadata, scale=1.0, instance_mode=ColorMode.SEGMENTATION)
v = v.draw_instance_predictions(filtered_predictions)


原文地址:https://blog.csdn.net/weixin_41012767/article/details/142421162

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!