search

KidsWithTokens / MedSegDiff

Medical Image Segmentation with Diffusion Model
MIT License
976 stars 145 forks source link

code #148

Open 1390806607 opened 6 months ago

1390806607 commented 6 months ago 
    res = torch.where(mask > 0, 1, 0)   #merge all tumor classes into one to get a binary segmentation mask

Hello, I don't quite understand why all categories of masks are merged into two categories of masks.Can you explain it to me?