https://github.com/victoorv/mri_segmentation

U-Net for MRI segmentation.
https://github.com/victoorv/mri_segmentation

computer-vision data-science image-processing image-segmentation kaggle-dataset machine-learning mri mri-brain mri-data mri-images mri-segmentation opencv pytorch segmentation u-net u-net-image-segmentation u-net-segmentation

Last synced: 2 months ago
JSON representation

U-Net for MRI segmentation.

• Host: GitHub
• URL: https://github.com/victoorv/mri_segmentation
• Owner: VictoorV
• Created: 2025-01-05T18:44:20.000Z (4 months ago)
• Default Branch: main
• Last Pushed: 2025-01-17T22:09:03.000Z (4 months ago)
• Last Synced: 2025-01-17T23:20:26.967Z (4 months ago)
• Topics: computer-vision, data-science, image-processing, image-segmentation, kaggle-dataset, machine-learning, mri, mri-brain, mri-data, mri-images, mri-segmentation, opencv, pytorch, segmentation, u-net, u-net-image-segmentation, u-net-segmentation
• Language: Jupyter Notebook
• Homepage: https://www.kaggle.com/datasets/mateuszbuda/lgg-mri-segmentation
• Size: 30.7 MB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# U-Net for MRI segmentation

U-Net implementation in PyTorch for MRI segmentation

## Data

Dataset from Kaggle : [kaggle.com/mateuszbuda/lgg-mri-segmentation](https://www.kaggle.com/mateuszbuda/lgg-mri-segmentation).

## Results

|![animation](https://github.com/user-attachments/assets/2dd4bb44-46aa-480d-a460-b1d63c576611)|

|:-------:|

| 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](https://github.com/user-attachments/assets/6d9f53b8-55e3-4bc0-9ef8-4d65a4c55d95)