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

Y_Hungry

已于 2024-09-21 18:07:30 修改

阅读量193

点赞数 3

文章标签： python 开发语言

于 2024-09-21 18:05:45 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_41012767/article/details/142421162

版权

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)