U-Net Brain MRI Segmentation

Implementation of 2D U-Net Segmentation model from scratch using PyTorch. The following model was implemented.

Install dependencies

Install libraries in your virtual python environment (e.g. conda) pip install -r requirements.txt

Training the model

Info about training config parameters: python main.py --help

E.g. python main.py --max-epochs 10 --batch-size 16 --device "cuda" --experiment-name "test_experiment"

Segmentation results