ViT3D-BrainTumorSegmentation
ViT3D-BrainTumorSegmentation copied to clipboard
Segmentation of Brain Tumors using Vision Transformer
Vision Transformer
Spread sheet including Results and related works is available in here.
Dataset
Dataset is available in here.
- Training: 3D MRI images from
369
patients (Height=240, Width=240, Depth=155)- Images: 4 modalities
- Masks: 4 classes(0, 1, 2, and 4)
- Validation: 3D MRI images from
125
patients (Height=240, Width=240, Depth=155)- Images: 4 modalities
Models
- SwinUNETR
2. UNETR
3. UNET
Metrics
- Dice Loss
Requirements
- git+https://github.com/Project-MONAI/MONAI#[email protected]+271.g07de215c
- torch
- nibabel
- tqdm
- einops
- numpy
- SimpleITK
Install the requirements using command below:
pip install -r requirements.txt
Run
python scripts.main --model swinunetr --epochs 10 --batch_size 2