HiLab-git / WSL4MIS

Scribbles or Points-based weakly-supervised learning for medical image segmentation, a strong baseline, and tutorial for research and application.
MIT License
205 stars 34 forks source link

issue of the MumfordShah_Loss #8

Closed EnQing626 closed 2 years ago

EnQing626 commented 2 years ago

Thank you for your code. It seems that there is a small issue with the inputs of the Mumfordshah_loss. loss_mumfordshah = mumfordshah_loss(volume_batch, outputs_soft) should be modified to loss_mumfordshah = mumfordshah_loss(outputs_soft, volume_batch).