New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

question #13

Open

Huster-Hq opened this issue Jan 8, 2023 · 1 comment

Huster-Hq commented Jan 8, 2023

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

Owner

LiWentomng commented Jan 8, 2023

@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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment