https://github.com/strcoder4007/u-net-image-segmentation

MRI Image Segmentation using U-Net implementation in Tensorflow 2
https://github.com/strcoder4007/u-net-image-segmentation

cs230 dataset tensorflow2 unet unet-image-segmentation

Last synced: about 2 months ago
JSON representation

MRI Image Segmentation using U-Net implementation in Tensorflow 2

• Host: GitHub
• URL: https://github.com/strcoder4007/u-net-image-segmentation
• Owner: strcoder4007
• License: mit
• Created: 2024-03-19T16:44:28.000Z (10 months ago)
• Default Branch: main
• Last Pushed: 2024-03-28T14:19:39.000Z (9 months ago)
• Last Synced: 2024-03-28T15:39:44.945Z (9 months ago)
• Topics: cs230, dataset, tensorflow2, unet, unet-image-segmentation
• Language: Jupyter Notebook
• Homepage:
• Size: 453 KB
• Stars: 0
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# U-net architecture implementation for MRI image segmentation

Research Paper: https://arxiv.org/pdf/1505.04597.pdf

![U-Net Architecture](/images/unet.png)

## Dataset

https://huggingface.co/datasets/rainerberger/Mri_segmentation

## How to run

- Run ```python data.py``` to download the MRI dataset from hugging face into dataset directory

- Run ```python train.py``` to train the model and predict the image segmentations