question

[Huster-Hq]

请问boxlevelset在mmdet框架中测试，如何只输出掩膜图？就是不需要有原图的叠加

[LiWentomng]

@Huster-Hq
你好，可以尝试修改 mmdet/core/visualization/image.py 中的imshow_det_bboxes 函数。
关于输出segm的相关代码如下，

        if segms is not None:
            color_mask = mask_colors[labels[i]]
            mask = segms[i].astype(bool)
            img[mask] = img[mask] * 0.5 + color_mask * 0.5