OpenGVLab / SAM-Med2D

Official implementation of SAM-Med2D

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The code for visualizing the results

Lzy-dot opened this issue · comments

commented

Great job! Could you please provide the code for the visualization part?

In the 'save_masks' function, you can set whether to visualize the prompt by adjusting the 'visual_prompt' parameter. You can try setting 'visual_prompt' to True.