U-Net implementation in PyTorch for MRI segmentation
Dataset from Kaggle : kaggle.com/mateuszbuda/lgg-mri-segmentation.
![]() |
---|
87% Dice, 82% IoU |
(Display may be improved with SimpleITK)
The following graph shows the Dice and IoU scores for 5 randomly selected patients :