Skip to content

VictoorV/mri_segmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 

Repository files navigation

U-Net for MRI segmentation

U-Net implementation in PyTorch for MRI segmentation

Data

Dataset from Kaggle : kaggle.com/mateuszbuda/lgg-mri-segmentation.

Results

animation
87% Dice, 82% IoU

(Display may be improved with SimpleITK)

The following graph shows the Dice and IoU scores for 5 randomly selected patients :

image