Skip to content

Latest commit

 

History

History
61 lines (44 loc) · 1.44 KB

README.md

File metadata and controls

61 lines (44 loc) · 1.44 KB
Raw

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

Install the requirements using command below:

pip install -r requirements.txt

Run

python scripts.main --model swinunetr --epochs 10 --batch_size 2

References