transformers-for-segmentation

Transformers for 3d image segmentation

• Models

  • UnetR
  • UnetR + deformable-attention

• DRL framework : PyTorch

• Dataset : BTCV

Install

git clone https://github.com/isk03276/transformers-for-segmentation
cd transformers-for-segmentation
pip install -r requirements.txt

Getting Started

python main.py --batch-size [BATCH-SIZE] --model-name [MODEL-NAME(ex. 'unetr', 'patch-wise-deformable-attention')] --n-folds 5 #train
python main.py --test

Results

- BTCV

References

• UNETR: Transformers for 3D Medical Image Segmentation
• Vision Transformer with Deformable Attention