ViT3D-BrainTumorSegmentation icon indicating copy to clipboard operation
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.

  1. Training: 3D MRI images from 369 patients (Height=240, Width=240, Depth=155)
    • Images: 4 modalities
    • Masks: 4 classes(0, 1, 2, and 4)

training image

  1. Validation: 3D MRI images from 125 patients (Height=240, Width=240, Depth=155)
    • Images: 4 modalities

validation image

Models

  1. SwinUNETR

swinunetr 2. UNETR

unetr 3. UNET

unet

Metrics

  1. 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

References