How to create a visualization similar to Figure 4 in the paper.

wurenkai / UltraLight-VM-UNet

[arXiv] The official code for "UltraLight VM-UNet: Parallel Vision Mamba Significantly Reduces Parameters for Skin Lesion Segmentation".

235 stars 33 forks source link

How to create a visualization similar to Figure 4 in the paper. #19

Closed cpy0029 closed 7 months ago

cpy0029 commented 7 months ago

How do I create a visualization similar to Figure 4, including for models such as UltraLightVM-Unet, LightM-Unet, VM-UNet, and others?

Thanks!!!

wurenkai commented 7 months ago

Hi, our model generates visualized segmentation plots (original at the top, Ground Truth in the middle, and predicted at the bottom) in 'outputs' in the 'results' folder after both training and testing. And the formation of Figure 4 requires some post-processing (cut and merge) to form.

cpy0029 commented 7 months ago

Thanks for your answer!!!